题意:给你一些数,把这些数分成M组,每组的花费是这组的 (max-min)^2。求最小花费
思路:
斜率优化+二维DP
f[i][m] 表示将前i个分作m个集合所得最小消费
首先应该排序,假设1,2,3,5,4 第四个数是5,花费一定比是4大。【贪心】
f[i][m] = min{f[j][m-1]+(a[i]-a[j+1])*(a[i]-a[j+1])}; 排序是为了让a[i]-a[j+1]总是最小的
若有j>k,且决策j优于决策k则有:
f[j][j-1]-f[k][j-1]+sq(a[j+1])-sq(a[k+1]) <= 2*(a[j+1]-a[k+1])*a[i]
先进行m循环枚举f[][m],每一层维护一个单调队列即可。
乘除耗费时间悬殊,如果直接除这个题就超时了。
代码:
1 #include <bits/stdc++.h> 2 using namespace std; 3 typedef long long ll; 4 #define mem(a) memset(a,0,sizeof(a)) 5 #define mp(x,y) make_pair(x,y) 6 const int INF = 0x3f3f3f3f; 7 const ll INFLL = 0x3f3f3f3f3f3f3f3fLL; 8 inline ll read(){ 9 ll x=0,f=1;char ch=getchar(); 10 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 11 while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} 12 return x*f; 13 } 14 ////////////////////////////////////////////////////////////////////////// 15 const int maxn = 1e4+10; 16 int n,M; 17 int a[maxn],f[maxn][5005],q[maxn]; 18 int getup(int j,int k,int m){ 19 return f[j][m-1]-f[k][m-1]+a[j+1]*a[j+1]-a[k+1]*a[k+1]; 20 } 21 22 int getdown(int j,int k,int m){ 23 return 2*(a[j+1]-a[k+1]); 24 } 25 26 int main(){ 27 int T=read(); 28 for(int cas=1; cas<=T; cas++){ 29 n=read(),M=read(); 30 for(int i=1; i<=n; i++) 31 a[i]=read(); 32 sort(a+1,a+1+n); 33 for(int i=1; i<=n; i++) f[i][1] = (a[i]-a[1])*(a[i]-a[1]); 34 35 for(int m=2; m<=M; m++){ 36 int L=0,R=0; 37 for(int i=1; i<=n; i++){ 38 while(L<R && getup(q[L+1],q[L],m) <= a[i]*getdown(q[L+1],q[L],m)) L++; 39 int j = q[L]; 40 f[i][m] = f[j][m-1]+(a[i]-a[j+1])*(a[i]-a[j+1]); 41 while(L<R && getup(i,q[R],m)*getdown(q[R],q[R-1],m) <= getup(q[R],q[R-1],m)*getdown(i,q[R],m)) R--; 42 q[++R] = i; 43 } 44 } 45 46 cout << "Case " << cas << ": " << f[n][M] << endl; 47 } 48 49 return 0; 50 }
来源:https://www.cnblogs.com/yxg123123/p/6827708.html