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;
}