hdu 3507 斜率优化dp

简单的斜率优化 但是要注意数据中有0，要特殊处理一下。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=5e5+9;
long long sum[maxn],dp[maxn];
int que[maxn];

bool chk1(int k,int j,int i)
{
    return (dp[j]-dp[k]+sum[j]*sum[j]-sum[k]*sum[k])<=2*sum[i]*(sum[j]-sum[k]);
}

bool chk2(int k,int j,int i)
{
    if(sum[i]==sum[j]) return true;
    long long a=dp[j]-dp[k]+sum[j]*sum[j]-sum[k]*sum[k];
    a*=(sum[i]-sum[j]);
    long long b=dp[i]-dp[j]+sum[i]*sum[i]-sum[j]*sum[j];
    b*=(sum[j]-sum[k]);
    return a>b;
}

int main()
{
    freopen("in.txt","r",stdin);
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        sum[0]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&sum[i]);
            sum[i]+=sum[i-1];
        }
        int front=1,end=0;
        que[++end]=0;
        dp[0]=0;
        for(int i=1;i<=n;i++)
        {
            while(front<end&&chk1(que[front],que[front+1],i))
            front++;
            int j=que[front];
            dp[i]=dp[j]+(sum[i]-sum[j])*(sum[i]-sum[j])+m;
            while(front<end&&chk2(que[end-1],que[end],i))
            end--;
            que[++end]=i;
        }
        printf("%lld\n",dp[n]);
    }
    return 0;
}