BZOJ P1911:[Apio2010]特别行动队

斜率优化DP第三发

下面是代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std;
long long f[1000003],num[1000003];
long long sum[1000003];
long long n,a,b,c;
int q[1000003];
long long M(long long x){return x*x;}
double xie(long long j,long long k){
	return((f[k]-f[j]+a*(M(sum[k])-M(sum[j]))+b*(sum[j]-sum[k])+0.0)/2.0/(sum[k]-sum[j])/a);
}
int main(){
	scanf("%lld%lld%lld%lld",&n,&a,&b,&c);
	for(int i=1;i<=n;i++){
		scanf("%lld",&num[i]);
	}
	for(int i=1;i<=n;i++){
		sum[i]=sum[i-1]+num[i];
	}
	f[0]=0;
	int h=1,t=1;
	q[1]=0;
	for(int i=1;i<=n;i++){
		while(h<t&&xie(q[h],q[h+1])<=sum[i]){
			h++;
		}
		int j=q[h];
		f[i]=f[j]+a*(M(sum[j]-sum[i]))+b*(sum[i]-sum[j])+c;
		while(h<t&&xie(q[t],q[t-1])>xie(i,q[t])){
			t--;
		}
		q[++t]=i;
	}
	printf("%lld",f[n]);
	return 0;
}
/*
in:
4
-1 10 -20
2 2 3 4
out:
9
*/