算法练习题 4（水龙头接水问题）

 

我：

#include <iostream>

using namespace std;
/*
n和m 表示的是接水人数 和水龙头个数
第二行 n个正整数 表示的是 w1,w2……wn 表示i号同学的接水量
输出只有一行 表示 接水所需的总时间
如：
5 3
4 4 2 1
输出 4
1<=n<=10000,1<=m<=100;
1<=wi<=100.
*/
int getMinIndex(int m, int wm[]) {

	int minW = 1;
	for (int i = 1; i <= m; i++)
	{
		int temp = wm[i];
		if (temp < wm[minW]) {
			minW = i;
		}

	}
	return minW;

}
int main() {
	int n=0, m=0;//n接水人数，m水龙头
	int w[105] = {0};//接水量
	cin >> n>>m;
	for (int i = 1; i <=n; i++)
	{
		cin >> w[i];
	}

	int maxW = 0;
	int wm[105] = { 0 };//1到m个水龙头分别的接水量
	for (int i = 1; i <= n; i++)
	{
		//初始化赋值
		if (i <= m) {;
			wm[i] = w[i];
		}else{
		//始终找出最小的值
		int minIndex = getMinIndex(m, wm);
		wm[minIndex] = wm[minIndex] + w[i];
		}

	}

	for (int i = 1; i <= m; i++)
	{
		int temp = wm[i];
		if (temp > maxW) {
			maxW = wm[i];
		}

	}

	cout << maxW << endl;



	return 0;
}

 

 

大佬：

/*
	数组 w 而言:
		下标: 数组下标去表示每个人的编号
		元素值: 表示接水量
			w[1] = 5  --> 1 号的接水量是 5
	关于数组 ar
		下标: 表示水龙头的编号
		元素值: 去表示该水龙头上消耗的时间
			ar[1] = 4   --> 1 号水龙头的供水量是 4，即消耗了 4 秒钟
*/


int w[10001] = {0}, ar[101] = {0};
	int n, m, max = 0;

	cin >> n >> m;
	// 得到每个人的接水量
	for(int i = 1; i <= n; i++) {
		cin >> w[i];
	}
	
	// 先派 m 个人去接水
	for(int i = 1; i <= m; i++) {
		ar[i] = w[i];   // 第 i 号同学需要接多少水，那么第 i 号水龙头消耗多长时间
	}

	// 编号 m+1  -- n 的同学在排队
	for(int i = m+1; i <=n; i++) {
		// 从 1 号水龙头开始扫描
		int k = 1;   // 起始时，代表 1 号水龙头
		for(int j = 2; j <= m; j++) {   // 扫描水龙头，判断哪个水龙头最先空出来
			if(ar[j] < ar[k]) {
				k = j;
			}
		}
		
		// 此时，找出耗时最小的第 k 号水龙，意味着这个水龙头上会有人补进来
		// 第 i 号同学补到第 k 个水龙头
		ar[k] = ar[k] + w[i];
	}
	
	for(int i = 1; i <= m; i++) {
		if(ar[i] > max) {
			max = ar[i];
		}
	}

	cout << max << endl;