bzoj 4518

4518

思路：

　　斜率优化；

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define maxn 3005
#define ll long long
ll s[maxn],q[maxn],f[maxn][maxn];
bool getx(int i,int x1,int x2,int x3)
{
    return (-s[x2]*s[x2]-f[i-][x2]+s[x1]*s[x1]+f[i-][x1])*
    (*s[x2]-*s[x3])>(-s[x3]*s[x3]-f[i-][x3]+s[x2]*s[x2]+
    f[i-][x2])*(*s[x1]-*s[x2]);
}
ll getans(int i,int v,int k)
{
    return -*s[v]*s[k]+s[k]*s[k]+f[i-][k]+s[v]*s[v];
}
int main()
{
    int n,m;scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++) scanf("%lld",&s[i]),s[i]+=s[i-];
    memset(f,,sizeof(f)),f[][]=;
    for(int i=;i<=m;i++)
    {
        int head=,tail=;
        for(int v=;v<=n;v++)
        {
            while(head+<=tail&&getx(i,q[tail-],q[tail-],q[tail]))
                q[tail-]=q[tail],tail--;
            while(head<tail&&getans(i,v,q[head])>=getans(i,v,q[head+])) head++;
            f[i][v]=getans(i,v,q[head]),q[++tail]=v;
        }
    }
    printf("%lld\n",f[m][n]*m-s[n]*s[n]);
    return ;
}