刷题系列

电梯问题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 层
"""