接水问题 python

解题思路:先把学生里面前m个导入到数组中，然后通过k计算接水量，s即总人数减去除了开头接水人数的剩余人数，最后输出接水列表最大值

注意事项:示例过大很容易超时所以直接设置列表长度为10000避免超出列表范围

参考代码:

n,m=map(int,input().split())

stu=list(map(int,input().split()))+[0]*10000

k=0

s=0

if n<m:

    m=n

w=[0]*10000

for i in range(m):

    w[i]=stu[i]

while m+s<n:

    k+=1

    for i in range(m):

        if w[i]<=k:

            w[i]+=stu[m+s]

            s+=1

print(max(w))