bzoj 1911: [Apio2010]特别行动队

Description

Input

Output

Sample Input

4
-1 10 -20
2 2 3 4

Sample Output

9

HINT

斜率优化DP

首先我们考虑初始转移方程f[i]=max(f[j]+a*(s[i]-s[j])^2+b*(s[i]-s[j])+c);其中s为x的前缀和

考虑j>k的时候j比k优

那么有f[j]+a*(s[i]-s[j])^2+b*(s[i]-s[j])+c<f[k]+a*(s[i]-s[k])^2+b*(s[i]-s[k])+c

移项后可得(f[j]+a*s[j]*s[j])-(f[k]+a*s[k]*s[k]))/(s[j]-s[k])<2*a*s[i]

维护下即可

【一开始不知为何把x排了个序，结果一直wa】

#include<cstdio>
#include<algorithm>
using namespace std;
long long x[1000001],s[1000001];
long long f[1000001];
int q[1000001];
long long a,b,c;
inline double solve(int k,int j)
{
     return double((f[j]+a*s[j]*s[j])-(f[k]+a*s[k]*s[k]))/double((s[j]-s[k]));
}
int main()
{
//	 freopen("commando.in","r",stdin);
//	 freopen("commando.out","w",stdout);
     int n;
     scanf("%d",&n);
     scanf("%lld%lld%lld",&a,&b,&c);
     int i;
     for(i=1;i<=n;i++)
          scanf("%lld",&x[i]);
    // sort(x+1,x+1+n);
     for(i=1;i<=n;i++)
          s[i]=s[i-1]+x[i];
     int l=1,r=0;
     r++;
     q[r]=0;
     for(i=1;i<=n;i++)
     {
          while(l<r&&solve(q[l],q[l+1])>double(((long long)2*a*s[i])+b))
               l++;
          f[i]=f[q[l]]+a*(s[i]-s[q[l]])*(s[i]-s[q[l]])+b*(s[i]-s[q[l]])+c;
               while(l<r&&solve(q[r-1],q[r])<solve(q[r],i))
                    r--;
          r++;
          q[r]=i;
     }
     printf("%lld\n",f[n]);
     return 0;
}