【寒假每日一题】luogu P1223排队接水

题目链接：P1223 排队接水 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

有 n 个人在一个水龙头前排队接水，假如每个人接水的时间为 Ti​，请编程找出这 n 个人排队的一种顺序，使得 n 个人的平均等待时间最小。

输入格式

第一行为一个整数 n。

第二行 n 个整数，第 i 个整数 Ti​ 表示第 i 个人的等待时间 Ti​ 。

输出格式

输出文件有两行，第一行为一种平均时间最短的排队顺序；第二行为这种排列方案下的平均等待时间（输出结果精确到小数点后两位）。

输入输出样例

输入 #1

10 
56 12 1 99 1000 234 33 55 99 812

输出 #1

3 2 7 8 1 4 9 6 10 5
291.90

说明/提示

n≤1000，ti ​≤ 10^6，不保证 ti​ 不重复。

当 ti​ 重复时，按照输入顺序即可（sort 是可以的）。

AC code 1：（常规解法）

#include<iostream>
#include<algorithm>

using namespace std;

int main()
{
    typedef long long ll;
	ll n;
	cin>>n;
	ll a[n],sum[n+1]={0},sum_=0;
	ll b[n];
	for(int i=0;i<n;i++)
	{
	    cin>>a[i];
	    b[i]=a[i]; // b[]数组用来保存未排序之前的原数组，以便于后面找到相应元素所对应的下标。
	}
	sort(a,a+n);
	for(int s=0;s<n-1;s++) // 类似于前缀和
	{
		sum[s+1]=a[s]+sum[s];
		sum_+=sum[s+1];
	}
	
	for(int s=0;s<n;s++) // 利用双重循环找到该值在原数组中（即未排序之前）所对应的下标。
	{
	    for(int s_=0;s_<n;s_++)
	    {
	        if(a[s]==b[s_])
	        {
	            b[s_]=-1; // 为了防止重复输出相同的下标，若该值已经输出过则赋值为-1。
	            cout<<s_+1<<' ';
	            break;
	        }
	    }
	}
	printf("\n%.2lf\n",sum_*1.0/n);
	
    return 0;
}

AC code 2：（STL解法）（利用 vector<pair<key,value>> 容器）

#include<iostream>
#include<algorithm>
#include<vector>

using namespace std;

/*
typedef pair<int,int> p;
 
bool cmp(p i,p j)
{
	return i.first < j.first;
}
*/ 

int main()
{
	typedef long long ll;
	ll n,sum_=0;
	cin>>n;
	
	vector<pair<ll,ll>> a;
	ll sum[n]={0};
	
	for(int i=1;i<=n;i++)
	{	
		ll x;
		cin>>x;
		a.emplace_back(x,i); // x 对应于键key, i对应于值value。
	}
	sort(a.begin(),a.end()); // 若不加第三个参数 cmp, 则默认对 pair中的第一个参数键key按升序排序。
	
/*	for(auto &i:a)
	{
		cout<<i.second<<' ';
	}
*/
	for(int o=0;o<n;o++)
		cout<<a[o].second<<' '; // 等价于 cout<<(a.begin()+o)->second<<' ';
	for(int s=0;s<n-1;s++) // 类似于前缀和
	{
		sum[s+1]=a[s].first+sum[s];
		sum_+=sum[s+1];
	}
	printf("\n%.2lf\n",sum_*1.0/n);
	
	return 0;
}

说明：这里为什么不用 map<key,value> 容器呢？

因为首先map只能对键key自动按照升序排序，并且当key中有重复元素时会自动去重。因此不能使用sort函数对map的值value进行排序。

所以，这里我们使用 pair<key,value> 和 vector 容器 可以很方便地根据需要和题目要求对键key或值value进行排序，并且是一一绑定的对应关系。

还有这里需要注意为了防止爆int，需要用 long long !!!

（当初一开始提交有2个WA，重新看了一下题目数据范围 n≤1000，ti ​≤ 10^6 ，下载了一下测试点发现在计算时可能会超过int范围。）所以一定要仔细审题！！！