bzoj 1911: [Apio2010]特别行动队

Posted 2020-06-24 xiyuedong

 1 #include<cstdio>
 2 #include<iostream>
 3 #define M 1000009
 4 #define ll long long
 5 using namespace std;
 6 int n,zhan[M],h,t;
 7 ll a,b,c,f[M],sum[M];
 8 double pai(int a1,int a2)
 9 {
10     double ss=(f[a1]-f[a2])/(1.0*a*(sum[a2]-sum[a1]));
11     return ss-sum[a2]-sum[a1];
12 }
13 int main()
14 {
15     scanf("%d%lld%lld%lld",&n,&a,&b,&c);
16     for(int i=1;i<=n;i++)
17       {
18         int a1;
19         scanf("%lld",&a1);
20         sum[i]=sum[i-1]+a1;
21       }
22     h=0;
23     for(int i=1;i<=n;i++)
24       {
25         for(;h<t&&pai(zhan[h],zhan[h+1])>=-b/(1.0*a)-2*sum[i];h++);
26         f[i]=f[zhan[h]]+a*(sum[zhan[h]]-sum[i])*(sum[zhan[h]]-sum[i])+b*(sum[i]-sum[zhan[h]])+c;
27         for(;h<t&&pai(zhan[t-1],zhan[t])<=pai(zhan[t],i);t--);
28         zhan[++t]=i;
29       }
30     printf("%lld\n",f[n]);
31     return 0;
32 }

普通的斜率优化DP。