BZOJ 1911: [Apio2010]特别行动队(斜率优化+DP）

题目链接

算是斜率优化的经典题吧，公式好像挺好就能推出来，特别注意要long long，写写 居然还超时，托了一段时间，看了AC大神的题解，发现中间处理写搓了，计算的时候要记忆化，各种细节要注意，各种WA和TLE终于完成这个渣代码。。。

还可以把求解时候再优化一下，这样1800+水过了。。。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define N  1000001
#define eps 1e-9
long long p[N],que[N];
long long a,b,c;
long long dp[N],sum[N];
long long fx[N];
long long f(int x)//记忆化一下
{
    if(fx[x] > 0)
        return fx[x];
    fx[x] = dp[x]+a*sum[x]*sum[x]-b*sum[x];
    return fx[x];
}
double slope(int x,int y)//计算斜率
{
    return (f(y)-f(x))*1.0/(sum[y]-sum[x]);
}
int main()
{
    int i,n,str,end;
    scanf("%d",&n);
    scanf("%lld%lld%lld",&a,&b,&c);
    for(i = 1; i <= n; i ++)
    {
        scanf("%lld",&p[i]);
        sum[i] = sum[i-1] + p[i];
    }
    str = end = 0;
    for(i = 1; i <= n; i ++)
    {
        while(str < end&&slope(que[str],que[str+1]) - 2*a*sum[i] > eps)//维护队列
            str ++;
        dp[i] = dp[que[str]] + a*(sum[i]-sum[que[str]])*(sum[i]-sum[que[str]])+b*(sum[i]-sum[que[str]])+c;//这里可以再优化一下。。
        while(str < end&&slope(que[end],i) - slope(que[end-1],que[end]) > eps)//维护队列
            end --;
        end ++;
        que[end] = i;
    }
    printf("%lld\n",dp[n]);
    return 0;
}