接水问题（蓝桥杯ACM训练）

题目描述

学校里有一个水房，水房里一共装有m个龙头可供同学们打开水，每个龙头每秒钟的供水量相等，均为1。现在有n名同学准备接水，他们的初始接水顺序已经确定。将这些同学按接水顺序从1到n编号，i号同学的接水量为wi。接水开始时，1到m号同学各占一个水龙头，并同时打开水龙头接水。当其中某名同学j完成其接水量要求wj后，下一名排队等候接水的同学k马上接替j同学的位置开始接水。这个换人的过程是瞬间完成的，且没有任何水的浪费。即j同学第x秒结束时完成接水，则k同学第x+1秒立刻开始接水。若当前接水人数n’不足m，则只有n’个龙头供水，其它m−n’个龙头关闭。现在给出n名同学的接水量，按照上述接水规则，问所有同学都接完水需要多少秒。

样例  1  说明
第1秒，3人接水。第1秒结束时，1、2、3号同学每人的已接水量为1，3号同学接完水，4号同学接替3号同学开始接水。
第2秒，3人接水。第2秒结束时，1、2号同学每人的已接水量为2，4号同学的已接水量为1。
第3秒，3人接水。第3秒结束时，1、2号同学每人的已接水量为3，4号同学的已接水量为2。4号同学接完水，5号同学接替4号同学开始接水。
第4秒，3人接水。第4秒结束时，1、2号同学每人的已接水量为4，5号同学的已接水量为1。1、2、5号同学接完水，即所有人完成接水。
总接水时间为4秒。
 

输入

第1  行2  个整数n  和m，用一个空格隔开，分别表示接水人数和龙头个数。  第2  行n  个整数w1、w2、……、wn，每两个整数之间用一个空格隔开，wi  表示i  号同  学的接水量。 

数据规模和约定
1  ≤  n  ≤  10000，1  ≤m≤  100  且m≤  n；
1  ≤  wi  ≤  100。

输出

输出只有一行，1  个整数，表示接水所需的总时间。 

样例输入

5 3
4 4 1 2 1
8 4
23 71 87 32 70 93 80 76

样例输出

4
163

解题思路:（Python解决）
题目说明是由先来的人打水，后来的人排队，所以可以使用先进先出的队列这种数据结构，用列表记录正在打水的人需要的打水量，先将打水的人全部入队，再根据水龙头空闲的数量出队进入打水的列表，如果存在水龙头空闲则出队，用列表记录正在打水的人还需要的打水量，如果需要打水量为零也就是已经打满水，则从列表中删除，再次从队列中出队并且进入打水的列表，直到队列和列表均为空

注意事项:
不可使用遍历列表使用del删除列表元素，会出现列表索引超出范围的问题，也不可只使用遍历列表使用remove的方法，这样会因为索引问题跳过下一个元素，也就是如果两个元素相邻且都为零，此时只可删除第一个元素，跳过第二个元素

需要将输入的打水量转换为整数

参考代码:

import queue

q = queue.Queue()  # 创建队列
a, b = map(int, input().strip().split())
people = input().split()
time = 0
free = b  # 空闲水龙头数量
water = []
for i in range(a):  # 所有人入队
   q.put(int(people[i]))
while not q.empty() or not len(water) == 0:  # 队列为空和正在打水人数为零才结束循环
   while free > 0 and not q.empty():  # 出现空闲水龙头则出队进入列表
       water.append(q.get())
       free = free - 1
   for i in range(len(water)):  # 正在打水的人也就是列表所有元素减一
       water[i] = water[i] - 1
   water = [i for i in water if i != 0]  # 把列表中不为零的元素重新进入列表，也就是删除列表中为零的元素
   free = b - len(water)  # 更新空闲水龙头数量
   time = time + 1
print(time)