bzoj 4518: [Sdoi2016]征途 斜率优化dp

题意

Pine开始了从S地到T地的征途。
从S地到T地的路可以划分成n段，相邻两段路的分界点设有休息站。
Pine计划用m天到达T地。除第m天外，每一天晚上Pine都必须在休息站过夜。所以，一段路必须在同一天中走完。
Pine希望每一天走的路长度尽可能相近，所以他希望每一天走的路的长度的方差尽可能小。
帮助Pine求出最小方差是多少。
设方差是v，可以证明，v×m^2是一个整数。为了避免精度误差，输出结果时输出v×m^2。
n<=3000

分析

斜率优化dp直接上即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib> 
#include<cstring>
#include<algorithm>
#define LL long long
#define N 3005
using namespace std;

int n,m,q[N];
LL sum[N],f[N][N];

LL sqr(LL x)
{
    return x*x;
}

LL get1(int x,int y,int j)
{
    return f[x][j-1]+sum[x]*sum[x]-f[y][j-1]-sum[y]*sum[y];
}

LL get2(int x,int y)
{
    return 2*sum[x]-2*sum[y];
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
    {
        scanf("%lld",&sum[i]);
        sum[i]=sum[i-1]+sum[i]*m;
    }
    LL c=sum[n]/m;
    for (int i=1;i<=n;i++)
        f[i][1]=sqr(sum[i]-c);
    for (int j=2;j<=m;j++)
    {
        int head=1,tail=1;q[1]=j-1;
        for (int i=j;i<=n;i++)
        {
            while (head<tail&&get1(q[head],q[head+1],j)>(sum[i]-c)*get2(q[head],q[head+1])) head++;
            f[i][j]=f[q[head]][j-1]+sqr(sum[i]-sum[q[head]]-c);
            while (head<tail&&get1(q[tail-1],q[tail],j)*get2(q[tail],i)>=get1(q[tail],i,j)*get2(q[tail-1],q[tail])) tail--;
            q[++tail]=i;
        }
    }
    printf("%lld",(LL)f[n][m]/m);
    return 0;
}