Codeforces1256E - Yet Another Division Into Teams dp

n个人的skill值 输出最小diff之和 分组数 所属组别
5
1 1 3 4 2

3 1
1 1 1 1 1

6
1 5 12 13 2 15

7 2
2 2 1 1 2 1

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define fi first
#define se second
const int maxn=1e6+5;
const int INF=0x3f3f3f3f;
ll n,q,f,flag;
ll dp[maxn],pre[maxn],ans[maxn];//dp[i]表示前i-1个人划分好后的最小d值和
string s,t;
struct node{
	ll v;
	ll p;
}a[maxn];

bool cmp(node a,node b){
	return a.v<b.v;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i].v;
		a[i].p=i;
	}
	sort(a+1,a+n+1,cmp);
	memset(dp,INF,sizeof dp);
	dp[1]=0;	//前0个 最小d之和 
	for(int i=1;i<=n;i++){
		for(int j=2;j<=4 && i+j<=n;j++){
			ll diff=a[i+j].v-a[i].v;
			if(dp[i+j+1]>dp[i]+diff){
				dp[i+j+1]=dp[i]+diff;
				pre[i+j+1]=i;  //i~i+j 相同组 
			}//记录方案，表示[i,i+j]这些人分为一组 i=4 p[7]=4 4-6 4
		}
	}
	ll root=n+1,cnt=1;
	while(root!=1){	//n~!0
		for(int i=root-1;i>=pre[root];i--){//[p[root], root-1]这些人为同一组
			ans[a[i].p]=cnt;
		}
		cnt++;
		root=pre[root];
	}
	cout<<dp[n+1]<<" "<<cnt-1<<endl;
	for(int i=1;i<=n;i++){
		cout<<ans[i];
		if(i!=n)
			cout<<" ";
	}
	cout<<endl;
	return 0;
}