问题 A: 深入浅出学算法043-排队打水

p.s.自用

题目描述

有N个人排队到1个水龙头去打水，他们装满水桶的时间为T1，T2，…，Tn为整数，应如何安排他们的打水顺序才能使他们花费的时间最少？

输入

输入分2行，第一个数为N表示打水人数
第二行有N个数，为每人的打水时间

输出

输出文件有两行，第一行为一种排队顺序，即1到n的一种排列；第二行为这种排列方案下的平均等待时间(输出结果精确到小数点后两位)。
如果有多种排队方式使得平均等待时间一样小，选择字典序最小的排队方式。

样例输入 Copy

10 
56 12 1 99 1000 234 33 55 99 812

样例输出 Copy

3 2 7 8 1 4 9 6 10 5
532.00

提示

n<=1000

ti<=1e6，不保证ti不重复 

参考答案

#include <iostream>
#include <algorithm>
#include <vector>
#include <iomanip> // 输出两位小数

using namespace std;

int main()
{
	// 输入
	int n; // 打水人数
	cin >> n;
	vector<int> time(n + 1, 0); // 每人的打水时间
	vector<int> no(n + 1, 0); // 序号
	for (int i = 1; i <= n; i++)
	{
		cin >> time[i];
		no[i] = i;
	}
	// 求解：排序-冒泡排序
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n - i; j++)
		{
			if (time[j] > time[j + 1])
			{
				int x = time[j];
				time[j] = time[j + 1];
				time[j + 1] = x;
				int y = no[j];
				no[j] = no[j + 1];
				no[j + 1] = y;
			}
		}
	}
	// 计算 & 输出
	long long all_time = 0;
	for (int i = 1; i <= n; i++)
	{
		all_time += time[i] * (n - i + 1);
		cout << no[i] << " ";
	}
	double res = all_time * 1.0 / n;
	cout << endl;
	cout << fixed << setprecision(2) << res << endl;
	return 0;
}