BZOJ4518: [Sdoi2016]征途

BZOJ4518

式子化简后，就是求$m*\sum_{i=1}^m{xi^2}-{sum[n]}^2$的最小值。
容易有$f[i][j]=min(f[i-1][k]+(sum[j]-sum[k])^2)$
然后很容易得出具有决策单调性，可以斜率优化。
。。为什么我连队列和$l,r$都要开二维。。并不知道一维怎么做的。。
反正题目不卡空间，递推就懒得滚动数组了，还跑得慢。

【代码】

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <stack>
#include <cstring>
#include <vector>
#include <queue>
#define N 3005
#define M 200005
#define INF 0x7fffffff
using namespace std;
typedef long long ll;
typedef pair<int,int> pa;

ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
} 

int n,m,now;
int q[2][N],l[2],r[2];
ll sum[N],f[N][N];

ll Sqr(ll x){
    return x*x;
}

double Slope(int i,int j,int k){
    return (double)(f[i][k]+Sqr(sum[k])-f[i][j]-Sqr(sum[j]))/(double)(sum[k]-sum[j]);
}

int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) sum[i]=sum[i-1]+read();
    for(int i=1;i<=m;i++)
    {
        now^=1;l[now]=r[now]=0;
        for(int j=i;j<=n;j++)
        {
            while(l[now^1]<r[now^1]&&Slope(i-1,q[now^1][l[now^1]],q[now^1][l[now^1]+1])<=sum[j]*2) l[now^1]++;
            int t=q[now^1][l[now^1]];
            f[i][j]=f[i-1][t]+Sqr(sum[j]-sum[t]);
            while(l[now]<r[now]&&Slope(i,q[now][r[now]],j)<Slope(i,q[now][r[now]-1],q[now][r[now]])) r[now]--;
            q[now][++r[now]]=j;
        }
    }
    printf("%lld\n",m*f[m][n]-Sqr(sum[n]));
    return 0;
}