试题 算法训练 接水问题

试题 算法训练 接水问题

问题描述
　　学校里有一个水房，水房里一共装有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 行2 个整数n 和m，用一个空格隔开，分别表示接水人数和龙头个数。 第2 行n 个整数w1、w2、……、wn，每两个整数之间用一个空格隔开，wi 表示i 号同 学的接水量。
输出格式
　　输出只有一行，1 个整数，表示接水所需的总时间。
　　
样例输入
5 3
4 4 1 2 1
样例输出
4

样例输入
8 4
23 71 87 32 70 93 80 76
样例输出
163

输入输出样例 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 ≤ n ≤ 10000，1 ≤m≤ 100 且m≤ n；
　　1 ≤ wi ≤ 100。

#include<bits/stdc++.h>
using namespace std; 
#define N 15
int n,m;//n个人，m个水龙头
int minn=INT_MAX,maxn=INT_MIN;
int a[10005];//存放每个人的水量
int num;//保存每次找到最小水龙头的编号
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=m+1;i<=n;i++){//对后面的m+1~n个接水量进行循环
		minn=INT_MAX;//minn用来找出最小水量的水龙头
		for(int j=1;j<=m;j++){//对m个水龙头进行遍历
			if(a[j]<minn){
				minn=a[j];
				num=j;//将最小水量的水龙头编号保存
			}
		}
		a[num]+=a[i];//将最小水量的龙头加上后续的同学的接水量
	}
	for(int i=1;i<=m;i++){//对m个龙头进行遍历
		if(a[i]>maxn){
			maxn=a[i];//找到最大水量的龙头，就是最长时间
		}
	}
	cout<<maxn;
	return 0;
}

思路
同学接水的顺序是已经排好的，而且m≤ n，所以接水的人数大于或者等于水龙头的个数。
用一个数组来存放每个同学的接水量，
比如：共有m个水龙头，最开始是a[1]到a[m]号的同学接水，那么当接水量最小的那个接完水后（假设此时a[0]号接水量最小），下一个同学（a[m+1]号）会在a[0]号，那就让a[0]=a[0]+a[m+1]，依次遍历m+1到n，每一次都将后序的接水量加在目前最小水量的龙头上，最后找到最长时间（也就是最大水量）的水龙头即可。

此题的b站的讲解视频推荐：接水问题