这道题因为序列顺序对结果没有影响
而是要求Max-Min的绝对值的平方,因此首先想到排序
而若元素有重复,只要不涉及到最大最小值就对问题无影响
因此将问题转化为:将n个元素分为M个子串,使得题中要求权值最小
则设DP[i]为前i个元素的最小值
DP[i]=min(DP[k]+(a[i]-a[k+1])^2)
则又是一脸要斜率优化的表情
设k<j且j更优
DP[j]+(a[i]-a[j+1])^2<DP[k]+(a[i]-a[k+1])^2
==>DP[j]+a[j+1]^2-DP[k]-a[k+1]^2<a[i]*2*(a[[j+1]-a[k+1])
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int SIZEN=10050;
const int SIZEM=5050;
int a[SIZEN],txt=1;
int dp[SIZEN][2],q[SIZEN];
int getup(int k,int j,int m){
return dp[j][m^1]-dp[k][m^1]-a[k+1]*a[k+1]+a[j+1]*a[j+1];
}
int getdown(int k,int j,int m){
return 2*(a[j+1]-a[k+1]);
}
int getval(int i,int j,int m){
return dp[j][m^1]+(a[i]-a[j+1])*(a[i]-a[j+1]);
}
void solve(){
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
sort(a+1,a+n+1);
int head,tail;
//for(int i=1;i<=n;i++) dp[i][0]=INF;
dp[0][0]=a[0]=0;
for(int i=1;i<=n;i++) dp[i][1]=(a[i]-a[1])*(a[i]-a[1]);
int cur=1;
for(int j=2;j<=m;j++){
head=tail=0;
q[tail++]=j-1;
cur^=1;
for(int i=j;i<=n;i++){
while(head+1<tail){
int dy=getup(q[head],q[head+1],cur);
int dx=getdown(q[head],q[head+1],cur);
if((LL)dy<(LL)a[i]*dx) head++;
else break;
}
dp[i][cur]=getval(i,q[head],cur);
while(head+1<tail){
int dy1=getup(q[tail-2],q[tail-1],cur);
int dx1=getdown(q[tail-2],q[tail-1],cur)/2;
int dy2=getup(q[tail-1],i,cur);
int dx2=getdown(q[tail-1],i,cur)/2;
if(dy1*dx2>=dy2*dx1) tail--;
else break;
}
q[tail++]=i;
}
}
printf("Case %d: %d\n",txt++,dp[n][cur]);
}
int main()
{
//freopen("data.in","r",stdin);
int _;
scanf("%d",&_);
while(_--) solve();
return 0;
}