c/c++学习——贪心

1348: 深入浅出学算法043-排队打水
题目描述
有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 <bits/stdc++.h>
#define ll long long
using namespace std;
int main(){
	ll n,a[1001],b[1001],flag=1;
	cin>>n;
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
	for(int i=1;i<=n;i++){
		cin>>a[i];
		b[i]=i;
	}
	ll i=1,temp,s=0;
	while(flag==1){
		flag=0;
		for(int j=1;j<=n-i;j++){
			if(a[j+1]<a[j]){
				temp=a[j+1];
				a[j+1]=a[j];
				a[j]=temp;
				temp=b[j];
				b[j]=b[j+1];
				b[j+1]=temp;
				flag=1;
			}
		}
		i++;
	}
	for(int i=1;i<=n;i++){
		s+=a[i]*(n-i+1);
		cout<<b[i]<<" ";
	}
	printf("\n");
	float ave=(float)s/(float)n;
	printf("%.2f\n",ave);
	return 0;
}