斜率优化专题4——bzoj 1911: [Apio2010] 特别行动队 题解

【原题】

1911: [Apio2010]特别行动队

Time Limit: 4 Sec   Memory Limit: 64 MB
Submit: 2134   Solved: 911

Description

Input

Output

Sample Input

4
-1 10 -20
2 2 3 4

Sample Output

9

HINT

【分析】只要跟着我前面的题目走，这道题真的是太水了。神马题解都不用参考，公式随便推。

易知方程是f[i]=max(f[j]+A*(sum[i]-sum[j])^2+B*(sum[i]-sum[j])+C)
设k比j优。
f[k]+A(sum[i]-sum[k])^2+B(sum[i]-sum[k])+C>f[j]+A(sum[i]-sum[j])^2+B(sum[i]-sum[j])+C
f[k]-2*A*sum[i]*sum[k]+A*sum[k]^2-B*sum[k]>f[j]-2*A*sum[i]*sum[j]+A*sum[j]^2-B*sum[j]
f[k]-f[j]+A*(sum[k]^2-sum[j]^2)+B*(sum[j]-sum[k])>2*A*sum[i]*(sum[k]-sum[j])
(f[k]-f[j]+A*(sum[k]^2-sum[j]^2)+B*(sum[j]-sum[k]))/2/(sum[k]-sum[j])/A<sum[i]

【代码】

#include<cstdio>
#include<algorithm>
using namespace std;
long long n,a[1000005],f[1000005],q[1000005],sum[1000005],ans,h,t,i,A,B,C,j;
long long M(long long x){return x*x;}
double xie(long long k,long long j)
{
  double temp=(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;
  return temp;
}
int main()
{
  scanf("%lld",&n);
  scanf("%lld%lld%lld",&A,&B,&C);
  for (i=1;i<=n;i++)
    scanf("%lld",&a[i]),sum[i]=sum[i-1]+a[i];
  f[0]=0;h=t=1ll;q[1]=0;
  for (i=1;i<=n;i++)
  {
    while (h<t&&xie(q[h+1],q[h])<=sum[i]) 
      h++;
    f[i]=f[q[h]]+A*M(sum[i]-sum[q[h]])+B*(sum[i]-sum[q[h]])+C;
    while (h<t&&xie(q[t],q[t-1])>xie(i,q[t])) t--;  
    q[++t]=i;  
  }  
  printf("%lld",f[n]);
  return 0;
}