头歌 426、打水

题目描述

N个人要打水，有M个水龙头，第i个人打水所需时间为Ti，请安排一个合理的方案使得所有人的等待时间之和尽量小。

输入输出格式

输入格式 第一行两个正整数N``M 接下来一行N个正整数Ti。 输出格式 最小的等待时间之和。（不需要输出具体的安排方案）

输入输出样例1

输入 7 3 3 6 1 4 2 5 7 输出 11 解释（可选） 一种最佳打水方案是，将N个人按照Ti从小到大的顺序依次分配到M个龙头打水。 　例如样例中，Ti从小到大排序为1，2，3，4，5，6，7，将他们依次分配到3个龙头，则去龙头一打水的为1，4，7；去龙头二打水的为2,5；去第三个龙头打水的为3,6。 　第一个龙头打水的人总等待时间 = 0 + 1 + (1 + 4) = 6 　第二个龙头打水的人总等待时间 = 0 + 2 = 2 　第三个龙头打水的人总等待时间 = 0 + 3 = 3 　所以总的等待时间 = 6 + 2 + 3 = 11

输入输出样例2

输入 4 2 4 3 1 6 输出 4

说明提示

1≤N≤1000 1≤M≤1000 1≤Ti≤1000

#include <bits/stdc++.h>
using namespace std;
//排序后 倒序分配水龙头   
struct man{
	int lt;//水龙头号 
	int flag;//标号--打水用时 
};
bool cmp(man a,man b){
	return a.flag<b.flag;
}
int main()
{
	int n;
	int t;
	cin>>n>>t;
	man a[10086];
	for(int i=1;i<=n;i++){
		cin>>a[i].flag;
	}
	sort(a+1,a+1+n,cmp);
	
	
	int k=1;
	for(int i=n;i>=1;i--){//分配水龙头 
		a[i].lt=k;
		k++;
		if(k>t)k=1;
	}
	
	
	 
	int wt[10086]={0};//每个水龙头的等待时间 
	int all=0;//总时间 
	for(int i=1;i<=n-t;i++){
		wt[a[i].lt]+=a[i].flag;//累加 
		all+=wt[a[i].lt];
	}
	
	cout<<all;
}