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

BZOJ 1911: [Apio2010]特别行动队

题目概述:

有n个士兵,其战斗力为x,某一段连续的士兵的修正战斗力为

题目分析:

1.先思考一下朴素的DP,
ans[i]表示到i点的最大修正战斗力,sum[i]表示从1到i的战斗力之和

O(n^2)的算法显然超时,开始式子推导,进行斜率优化.

2.开始式子的推导:
假设当前i号的决策,存在j比k更优(j < k),则有

因为a<0&&sum[j]-sum[k]<0,所以有

3.然后再维护一个斜率单调的队列即可.

代码:

#include<cstdio>
#include<iostream>
#include<algorithm>

using namespace std;

typedef long long ll;
const int maxn=1000000+10;

ll ans[maxn];
int sum[maxn],que[maxn],a,b,c;

ll sqr(int x)
{
    return (ll)x*x;
}

double slope(int j,int k)//j<k
{
    return (ans[j]-ans[k]+a*(sqr(sum[j])-sqr(sum[k]))-b*(sum[j]-sum[k]))/(2.0*a*(sum[j]-sum[k]));
}

int main()
{
    int n;
    scanf("%d%d%d%d",&n,&a,&b,&c);
    for(int x,i=1;i<=n;i++) {
        scanf("%d",&x);
        sum[i]=sum[i-1]+x;
    }
    int head=0,tail=0;
    for(int i=1;i<=n;i++) {
        while(head<tail&&slope(que[head],que[head+1])<sum[i]) head++;
        int j=que[head];
        ans[i]=ans[j]+(ll)a*(sum[i]-sum[j])*(sum[i]-sum[j])+b*(sum[i]-sum[j])+c;
        while(head<tail&&slope(que[tail-1],que[tail])>slope(que[tail],i)) tail--;
        que[++tail]=i;
    }
    printf("%lld",ans[n]);
    return 0;
}