CF238B Boring Partition

Solution

想要得到差最大，只能是最小的自己一组或者全部在一组，因为：如果拿另一个和最小的一组，最小值只有可能最小，最大值只有可能更大，反证完毕。

那么我们可以将 (a) 数组从小到大排序，那么现在 (a_1) 是 (min a) ， (a_n) 是 (max a) ，当全部在一组时， (ans=f(a_n,a_{n-1})-f(a_1,a_2)=a_n+a_{n-1}-a_1-a_2) ，但是当 (a_1) 自己一组时， (ans=max(a_n+a_{n-1},a_n+a_1+h)-min(a_1+a_2+h,a_2+a_3)) 。

代码

#include<bits/stdc++.h>

using namespace std;
const int N=1e5+10;
int n,h;
struct node{
    int x,id;
    int val;
}a[N];

bool cmp1(node x,node y){
	return x.x<y.x;
} 

bool cmp2(node x,node y){
    return x.id<y.id;
}

int main(){
    scanf("%d%d",&n,&h);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i].x),a[i].id=i,a[i].val=0;
    sort(a+1,a+n+1,cmp1);
    if(n==2) printf("0
1 1
");
    else{
        if(max(a[n].x+a[n-1].x,a[1].x+a[n].x+h)-min(a[1].x+a[2].x+h,a[2].x+a[3].x)<a[n].x+a[n-1].x-a[1].x-a[2].x) a[1].val=1;
        cout<<min(max(a[n].x+a[n-1].x,a[1].x+a[n].x+h)-min(a[1].x+a[2].x+h,a[2].x+a[3].x),a[n].x+a[n-1].x-a[1].x-a[2].x)<<endl;
        sort(a+1,a+n+1,cmp2);
        for(int i=1;i<=n;i++)
            printf("%d%c",a[i].val==0?2:1," 
"[i==n]);
    }
    return 0;
}