DP（斜率最佳化）

 

Print Article

悲劇。不是數組寫小就是小於符號寫反。。。

明天比賽，期待RP爆發！

 

 

 所謂的斜率最佳化就是對頭維持單調，隊尾維護凸點！

#include <iostream>

#include <cmath>

using namespace std;

 

const int N=500010;

__int64 dp[N];

__int64 sum[N],dia;

int Q[N];

int n,m;

 

__int64 cal(int c,int k)

{

    return dp[c]+(sum[k]-sum[c])*(sum[k]-sum[c])+m;

}

 

__int64 Y(int c)

{

    return dp[c]+sum[c]*sum[c];

}

 

int main()

{

    int i,j;

 

    while(scanf("%d %d",&n,&m)!=EOF)

    {

        sum[0] = 0;

        for(i=1;i<=n;i++)

        {

            scanf("%d",&dia);

            sum[i]=sum[i-1]+dia;

        }

        int head,tail;

        head=0;

        tail=-1;

        dp[0]=0;

        Q[++tail]=0;

        for(i=1;i<=n;i++)

        {

            while(tail-head>=1 && cal(Q[head],i)>=cal(Q[head+1],i) )

                head ++;

            dp[i] = cal(Q[head],i);

            while(tail-head>=1 && (sum[Q[tail]] - sum[Q[tail-1]])*(Y(i)-Y(Q[tail-1]))<=

                (sum[i]-sum[Q[tail-1]])*(Y(Q[tail]) - Y(Q[tail-1])) )

                tail --;

            Q[++tail] = i;

        }

        printf("%I64d/n",dp[n]);

    }

    return 0;

}