Python: 接水问题

目录

题目描述

输入描述

输出描述

输入输出样例

思路： 

参考代码：

---

题目描述

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

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

现在给出 n 名同学的接水量，按照上述接水规则，问所有同学都接完水需要多少秒。

输入描述

第 1 行 2 个整数 n 和 m用一个空格隔开，分别表示接水人数和龙头个数。

第 2 行 n 个整数 w1​、w2​、⋯、wn​，每两个整数之间用一个空格隔开，wi​ 表示 i 号同学的接水量。

输出描述

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

输入输出样例

示例 1

输入

5 3 
4 4 1 2 1 

输出

4

示例 2

输入

8 4 
23 71 87 32 70 93 80 76 

输出

163

思路： 

1、当接水人数小于水龙头数时，最大的接水需求量即为接水时间

2、当接水人数大于水龙头数时，超出的人一定是在第一轮接水需求的后面接水，所以先将第一轮的接水的需求量排序，存放在列表里，然后将多余的人的接水需求量累加在列表的起始元素位置，再次排序，位于列表最后面的即为接水所需时间

参考代码：

import os
import sys

n,m=map(int,input().split())     #n表示接水人数，m表示水龙头个数
list1=list(map(int,input().split()))
if m>=n:                         #若水龙头的个数大于接水人数
  print(max(list1))              #则输出接水量最大的那个即为接完水所需的秒数
else:
  a=[list1[i] for i in range(m)] #列表生成式，将列表list1中前 m 个元素生成到列表 a 中
  a.sort()                       #对列表 a 进行升序排序（从小到大）
  for i in range(n-m):           
    a[0]+=list1[m+i]             #在列表a的首位=a[0]+list1[m+1]
    a.sort()
  print(a[m-1])

map（）函数：

map() 会根据提供的函数对指定序列做映射。

第一个参数 function 以参数序列中的每一个元素调用 function 函数，返回包含每次 function 函数返回值的新列表。

语法

map() 函数语法：

map(function, iterable, ...)

参数

• function -- 函数
• iterable -- 一个或多个序列

 实例：

def square(x) :         # 计算平方数
...     return x ** 2
...
>>> map(square, [1,2,3,4,5])    # 计算列表各个元素的平方
<map object at 0x100d3d550>     # 返回迭代器
>>> list(map(square, [1,2,3,4,5]))   # 使用 list() 转换为列表
[1, 4, 9, 16, 25]
>>> list(map(lambda x: x ** 2, [1, 2, 3, 4, 5]))   # 使用 lambda 匿名函数
[1, 4, 9, 16, 25]