【python】字节跳动大学水房接水

题目：

#字节跳动大学有一个水房，水房里一共装有m 个龙头可供同学们打开水，每个龙头每秒钟的供水量相等，均为1。

#现在有n 名同学准备接水，他们的初始接水顺序已经确定。将这些同学按接水顺序从1到n 编号，i 号同学的接

#水量为wi。接水开始时，1 到m 号同学各占一个水龙头，并同时打开水龙头接水。当其中某名同学j 完成其接水

#量要求wj 后，下一名排队等候接水的同学k会马上接替j 同学的位置开始接水。这个换人的过程是瞬间完成的，

#且没有任何水的浪费。即j 同学第x 秒结束时完成接水，则k 同学第x+1 秒立刻开始接水。若当前接水人数n’

#不足m，则只有n’个龙头供水，其它m-n’个龙头关闭。现在给出n 名同学的接水量，按照上述接水规则，问所有同学都接完水需要多少秒。

#coding:utf-8

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

def find_min(time_end):
    """
    """
    index = time_end.index(min(time_end))
    return index

def update(water, min_index, time_end):
    """
    """
    time_end[min_index] = time_end[min_index] + water
    return time_end

def run():
    """
    """
    m = 3
    n = 5

    water_list = [4, 4, 1, 2, 1]
    time_end = [0 for i in range(m)]

    max_time_cost = 0
    for i in range(n):

       min_index = find_min(time_end)
       time_end  = update(water_list[i], min_index, time_end) 
    
    max_time_cost = max(time_end)
    print("max_time_cost is %s" % max_time_cost)

if __name__ == "__main__":
    run()