【题目】
link:http://www.lydsy.com/JudgeOnline/problem.php?id=4518
【题解】
去年省选的时候。。。我是一枚很弱很弱的蒟蒻。这道题使用了大爆搜,然后爆零了。。。
现在看来其实很水
你就各种推导
发现答案就等于m*∑(bi^2) -s[n]^2
其中bi表示划分之后第i段的权值和,s表示前缀和
那么你就要最小化∑(bi^2)
那就划分DP,f[i][j]表示把前j个数划分成i段的最小∑bi^2
则f[i][j]=min( f[i-1][k] + (s[j]-s[k])^2 ) k∈[i,j)
这样是O(N^3)
那你就斜率优化,用g[j]代表f[i-1][j],f[j]代表f[i][j]
则f[j]=(g[k]+s[k]^2) -2s[j]s[k] +s[j]^2,
由于s[j]是前缀和,所以单调增,那么斜率2s[j]就单调增,维护下凸壳
然后就O(N^2)过了
这道题一遍AC了我还是很惊讶的
【代码】
#include <cstdio>
#include <algorithm>
#define maxn 3010
#define ll long long
using namespace std;
ll f[2][maxn], s[maxn], N, M;
struct point{ll x, y;double k;}q[maxn];
ll read(ll x=0)
{
char c=getchar();
while(c<48 or c>58)c=getchar();
while(c>=48 and c<=58)x=x*10+c-48,c=getchar();
return x;
}
void init()
{
ll i, a;
N=read();M=read();
for(i=1;i<=N;i++)a=read(),s[i]=s[i-1]+a;
}
void work(ll *f, ll *g, ll i)
{
ll l=1, r=1, x, y, j;
q[r++]=(point){s[i-1],g[i-1]+s[i-1]*s[i-1],0};
for(j=i;j<=N;j++)
{
while(l<r-1 and 2*s[j]>q[l+1].k)l++;
f[j]=q[l].y-2*s[j]*q[l].x+s[j]*s[j];
x=s[j],y=g[j]+s[j]*s[j];
while(l<r-1 and double(y-q[r-1].y)/(x-q[r-1].x)<q[r-1].k)r--;
q[r]=(point){x,y,double(y-q[r-1].y)/(x-q[r-1].x)},r++;
}
}
int main()
{
ll i, j;
init();
for(i=1;i<=N;i++)f[1][i]=s[i]*s[i];
for(i=2;i<=M;i++)work(f[i&1],f[~i&1],i);
printf("%lld\n",f[M&1][N]*M-s[N]*s[N]);
return 0;
}