刷题系列

电梯问题1

"""
有一部电梯,有m个乘客,每个乘客都从一楼开始上电梯,电梯到达某一层楼后,电梯停止,所有乘客从该层爬楼到自己的目的层(目的层大于等于2)。
问:电梯停在哪一层楼,能保证这次乘坐电梯的所有乘客爬楼梯的层数之和最少。
例:list=【0,0,3,4,5,3,2,0,1】#list【i】表示到第i层的乘客数目
输出:电梯停在第4层,所有乘客爬层总数为21  这题该怎么写呢  
"""
""" 其实第5层是21 他这个题表述的有问题 他没把列表的下标跟楼层的关系说明白 下标4的话其实是第5层 """ lst = [0,0,3,4,5,3,2,0,1] # 停在第4层 stop_index = 4 sum = 0 # index:层数 # value:要去这个层的人数 for index,value in enumerate(lst,1): print(index,value) # 指定层不用管 if index == stop_index: continue # 总的爬梯数 if index < stop_index: sum += value * (stop_index - index) elif index > stop_index: sum += value * (index - stop_index) print("sum>>>>>>",sum) print(3+5+6+6+5)

另外一种下标为0的解决方法

lst = [0,0,3,4,5,3,2,0,1]

total = len(lst)
print(f"一共有{total}层")
num = {}

for i in range(total):
    out = 0
    for j in range(total):
        out += abs(j-i) * lst[j]
    num[i] = out
    print(f"假如停在 {i} 层,剩下的人需要爬 {out} 层")

res = sorted(num.items(),key=lambda k:k[1])
print(f"最佳停的层数为 {res[0][0]},剩下的人需要爬 {res[0][1]} 层")
"""
一共有9层
假如停在 0 层,剩下的人需要爬 73 层
假如停在 1 层,剩下的人需要爬 55 层
假如停在 2 层,剩下的人需要爬 37 层
假如停在 3 层,剩下的人需要爬 25 层
假如停在 4 层,剩下的人需要爬 21 层
假如停在 5 层,剩下的人需要爬 27 层
假如停在 6 层,剩下的人需要爬 39 层
假如停在 7 层,剩下的人需要爬 55 层
假如停在 8 层,剩下的人需要爬 71 层
最佳停的层数为 4,剩下的人需要爬 21 层
"""

 

posted on 2020-05-31 08:49  江湖乄夜雨  阅读(155)  评论(0编辑  收藏  举报