简单化简一下可以发现,d这个题要我们求的其实是
m
∗
∑
i
=
1
m
s
2
−
s
u
m
2
m*\sum_{i=1}^{m}s^{2}-sum^{2}
m∗∑i=1ms2−sum2,注意到这里的m和sum其实都是常量,所以我们只要求出每一段的和平方的最小值就行了。然后这个是一个经典的决策单调的问题。
对于每一次转移,我们直接暴力求出mid点的转移过来的那个点,然后对两边分治就可以了,分治优化要比斜率优化要好写很多,当然斜率优化也要会做。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
const int N=3e3+7;
int a[N];
ll sum[N];
ll dp[N][N];
ll sq(ll x) { return x*x; }
void solve(int l,int r,int L,int R,int now) {
if(l>r||L>R) return;
int mid=(l+r)>>1;
int value=2e9;
int pos=0;
for(int i=L;i<mid&&i<=R;i++) {
if( sq(sum[mid]-sum[i])+dp[i][now-1]<value ) {
value=sq(sum[mid]-sum[i])+dp[i][now-1];
pos=i;
}
}
dp[mid][now]=value;
solve(l,mid-1,L,pos,now);
solve(mid+1,r,pos,R,now);
}
int main() {
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
dp[0][0]=0;
for(int i=1;i<=n;i++) dp[i][0]=2e9;
for(int i=1;i<=m;i++) solve(1,n,0,n,i);
printf("%lld\n",1LL*m*dp[n][m]-sum[n]*sum[n]);
return 0;
}