CodeForcess--598(div3)E. Yet Another Division Into Teams

原网址

http://codeforces.com/contest/1256/problem/E

大意

学校最多有2e5个学生，每个学生都有一个实力值，学校决定把他们组成队伍去参加比赛，每个队伍至少三个人，每个队伍的实力差为实力最高的减去最低的。为了避免实力差太过巨大，请你把它们分成最优的小组使所有队伍实力差的总和最小化。输出时输出实力差之和（res）与队伍总数（k），以及每个学生是哪一队的。

分析

首先明确两点。
一、分法一定是排序后连续的几名队员。
二、分的队伍越多则res越小。例如一个六人队伍肯定不如两个三人队伍。
由此得到一个推论，队伍只能是3，4，5个人的。
那么可以得到递推方程
$dp[i]=dp[i-d]-a[i-d+1].v+a[i].v$

#include<bits/stdc++.h>
using namespace std;
const int MAXN=2e5+30;
const int INF=0x3f3f3f3f;
struct nu{
	long long v;
	int num;
}a[MAXN];
int n,len[MAXN],team[MAXN];
long long dp[MAXN];
bool cmp(nu a,nu b){
	return a.v<b.v;
}
int main()
{
	while(~scanf("%d",&n)){
		for(int i=1;i<=n;i++){
			scanf("%lld",&a[i].v);
			a[i].num=i;
			dp[i]=INF;
		}
		sort(a+1,a+n+1,cmp);
		memset(len,0,sizeof(len));
		for(int i=1;i<=n;i++){
			for(int d=3;d<=min(i,5);d++){
				if(dp[i]>dp[i-d]-a[i-d+1].v+a[i].v){
					dp[i]=dp[i-d]-a[i-d+1].v+a[i].v;
					len[i]=d;
				}		
			}
		}
		int k=1;
		for(int i=n;i>0;i--){
			for(int j=i;j>i-len[i];j--){
				team[a[j].num]=k;
			}
			i=i-len[i]+1,++k;
		}
		printf("%lld %d\n",dp[n],k-1);
		for(int i=1;i<=n;i++){
			printf("%d%c",team[i],i==n?'\n':' ');
		}
	}
	return 0;
}