题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=4518
题目大意:
相当于:n个数字,将其分成m段,每段的数字和为该段的值,使方差最小..
===========================================
题解:
斜率优化【mdzz啊(口亨一定是题目表述不清!!!搞了我那么久
因为分成的段数的固定的,那么平均数sm也是固定的(这里的sum为总和)
所以方程很容易写到:
f[k][i]=f[k-1][j]+(sum[i]-sum[j]-sm)^2;
(设f[k][i]为以i为结尾已经分成了k段,sum[i]为前缀和)
-> 2*(sum[i]-sm)*sum[j]+f[k][i]=f[k-1][j]+sum[j]^2 +(sum[i]-sm)^2;
于是截距式就出来了,斜率优化QwQmd自己好zz
因为答案要乘上m^2,所以把m^2直接乘丢到式子里,即将所有数都先*m,最后再除个m,化简过程可以自己试试。
习惯了这种二维搞滚动orz不搞应该也行....吧日常LL咯~(我已经不管究竟要不要了= =反正又不会爆233
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
#define maxn 31000
int l,r,q[maxn],t;
LL sum[maxn],a[maxn],f[2][maxn];
LL sqrr(LL x){return x*x;}
double X(int j){return sum[j];}
double Y(int j){return f[1-t][j]+sqrr(sum[j]);}
double slop(int j1,int j2) {return (Y(j2)-Y(j1))/(X(j2)-X(j1));}
int main()
{
int n,m,i,k;LL sm;
sm=sum[0]=0;
scanf("%d%d",&n,&m);
for (i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
sm+=a[i];a[i]*=m;
sum[i]=sum[i-1]+a[i];
}t=1;
memset(f,63,sizeof(f));f[0][0]=0;
for (i=1;i<=n;i++) f[0][i]=sqrr(sum[i]-sm);
for (k=1;k<=m;k++)
{
l=r=1;q[l]=0;
memset(f[t],63,sizeof(f[t]));
for (i=1;i<=n;i++)
{
while (l<r && slop(q[l],q[l+1])<(sum[i]-sm)*2) l++;
int j=q[l];
f[t][i]=f[1-t][j]+sqrr(sum[i]-sum[j]-sm);
while (l<r && slop(q[r-1],q[r])>slop(q[r],i)) r--;
q[++r]=i;
}t=1-t;
}printf("%lld\n",f[1-t][n]/m);
return 0;
}