接水问题——蓝桥python
算法训练 接水问题
Description
学校里有一个水房,水房里一共装有m 个龙头可供同学们打开水,每个龙头每秒钟的 供水量相等,均为1。 现在有n 名同学准备接水,他们的初始接水顺序已经确定。将这些同学按接水顺序从1 到n 编号,i 号同学的接水量为wi。接水开始时,1 到m 号同学各占一个水龙头,并同时打 开水龙头接水。当其中某名同学j 完成其接水量要求wj 后,下一名排队等候接水的同学k 马上接替j 同学的位置开始接水。这个换人的过程是瞬间完成的,且没有任何水的浪费。即 j 同学第x 秒结束时完成接水,则k 同学第x+1 秒立刻开始接水。若当前接水人数n’不足m, 则只有n’个龙头供水,其它m−n’个龙头关闭。 现在给出n 名同学的接水量,按照上述接水规则,问所有同学都接完水需要多少秒。
Input
输入描述:
第1 行2 个整数n 和m,用一个空格隔开,分别表示接水人数和龙头个数。 第2 行n 个整数w1、w2、……、wn,每两个整数之间用一个空格隔开,wi 表示i 号同 学的接水量。
输入样例:
5 3
4 4 1 2 1
Output
输出描述:
输出只有一行,1 个整数,表示接水所需的总时间。
输出样例:
4
代码
下面展示具体代码
。
//接水问题
n,m=map(int,input().split())//n为接水人数、m为水龙头数
l=list(map(int,input().split()))//输入一组由空格分割的整数,并存放到列表中
if n<=m://人数少于水龙头数时
t=max(l)
else:
a=[l[i] for i in range(m)]//列表生成式,将l列表中的前m项存到a列表中
a.sort()//升序排序
for i in range(n-m):
a[0]+=l[m+i]//依次将未接水加接水时间最小值上
a.sort()//每次加完后排序
t=a[m-1]//列表最后一位即为接水总时间
print(t)
知识小结——map()函数
map() 会根据提供的函数对指定序列做映射。
第一个参数 function 以参数序列中的每一个元素调用 function 函数,返回包含每次 function 函数返回值的新列表
// map() 函数语法:
map(function, iterable, ...)