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;
}