codeforces572D. Minimization

传送门:http://codeforces.com/problemset/problem/572/D

思路:首先可以根据下标对k的模将他们分成互不相关的k组,组内可以贪心地使元素按大小顺序排,然后这组的贡献即为max-min

那我们就把a数组排序,那么每组元素一定是连续的。

那么题目就变为把有序数组a分为k组,其中第一种有n%k组,长度为(n/k+1),第二种有n-n%k组,长度为n/k

于是就可以DP了

设f[i][j]表示第一种用了i组,第二种用了j组的最小值,转移很简单,见代码。


#include<cstdio>
#include<cstring>
#include<algorithm>
#define abs(a) ((a)>0?(a):-(a))
const int maxk=5010,maxn=400010;
using namespace std;
typedef long long ll;
ll n,k,a[maxn],n1,n2,l1,l2;ll ans,f[maxk][maxk];
//f[i][j]表示第一种(长度多1的)用i次,第二种用j次

int main(){
	scanf("%I64d%I64d",&n,&k),l1=n/k+1,l2=n/k,n1=n%k,n2=k-n%k;
	for (ll i=1;i<=n;i++) scanf("%I64d",&a[i]);
	sort(a+1,a+1+n),memset(f,63,sizeof(f)),f[0][0]=0;
	for (ll i=0;i<=n1;i++)
		for (ll j=0;j<=n2;j++){
			ll pos=i*l1+j*l2;
			if (i) f[i][j]=min(f[i][j],f[i-1][j]+a[pos]-a[pos-l1+1]);
			if (j) f[i][j]=min(f[i][j],f[i][j-1]+a[pos]-a[pos-l2+1]);
		}
	printf("%I64d
",f[n1][n2]);
	return 0;
}


原文地址:https://www.cnblogs.com/thythy/p/5493513.html