apio2010特别行动队

http://www.elijahqi.win/archives/455
题目描述
你有一支由 n 名预备役士兵组成的部队，士兵从 1 到 n 编号，要将他们拆分 成若干特别行动队调入战场。出于默契的考虑，同一支特别行动队中队员的编号 应该连续，即为形如

的序列。 编号为 i 的士兵的初始战斗力为 xi ，一支特别行动队的初始战斗力 x 为队内 士兵初始战斗力之和，即

。
通过长期的观察，你总结出一支特别行动队的初始战斗力 x 将按如下经验公 式修正为

，其中 a, b, c 是已知的系数（a < 0）。 作为部队统帅，现在你要为这支部队进行编队，使得所有特别行动队修正后 战斗力之和最大。试求出这个最大和。 例如，你有 4 名士兵， 。经验公式中的参数为 a = –1, b = 10, c = –20。此时，最佳方案是将士兵组成 3 个特别行动队：第一队包含士兵 1 和士兵 2，第二队包含士兵 3，第三队包含士兵 4。特别行动队的初始战斗力分 别为 4, 3, 4，修正后的战斗力分别为 4, 1, 4。修正后的战斗力和为 9，没有其它 方案能使修正后的战斗力和更大。 输入输出格式 输入格式： 输入由三行组成。第一行包含一个整数 n，表示士兵的总数。第二行包含三 个整数 a, b, c，经验公式中各项的系数。第三行包含 n 个用空格分隔的整数 ，分别表示编号为 1, 2, …, n 的士兵的初始战斗力。 输出格式： 输出一个整数，表示所有特别行动队修正后战斗力之和的最大值。 输入输出样例 输入样例#1： 4 -1 10 -20 2 2 3 4 输出样例#1： 9 说明 20%的数据中，n ≤ 1000； 50%的数据中，n ≤ 10,000； 100%的数据中，1 ≤ n ≤ 1,000,000，–5 ≤ a ≤ –1，|b| ≤ 10,000,000，|c| ≤ 10,000,000，1 ≤ xi ≤ 100 dp方程：f[i]=max(f[j]+a*(sum[i]-sum[j])^2+b*(sum[i]-sum[j])+c) 如果j>k且j比k更优
f[j]-f[k]+a*sum[j]^2-a*sum[k]^2+b*(sum[k]-sum[j])>2*a*(sum[j]-sum[k])*sum[i]

#include<cstdio>
#include<queue>
#include<iostream>
#define N 1000010
using namespace std;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while  (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
long long s[N],f[N];
int  a,b,c;
deque<int>q;
inline double slope(int x,int y){
    return (double) (f[y]-f[x]+a*s[y]*s[y]-a*s[x]*s[x]+b*s[x]-b*s[y])/(2.0*a*(s[y]-s[x]));
}
int main(){
    //freopen("3628.in","r",stdin);
//    freopen("3628.out","w",stdout);
    int n=read(),tmp=0;s[0]=0;
    a=read(),b=read(),c=read();
    for (int i=1;i<=n;++i) tmp=read(),s[i]=s[i-1]+tmp;
    q.push_back(0);
    for (int i=1;i<=n;++i){
        while (q.size()>1&&slope(q[0],q[1])<s[i]) q.pop_front();
        tmp=q.front();int tmp2=s[i]-s[tmp],j;
        f[i]=f[tmp]+a*tmp2*tmp2+b*tmp2+c;
        while ((j=q.size())>1&&slope(q[j-1],i)<slope(q[j-2],q[j-1])) q.pop_back();
        q.push_back(i);
    }
    cout<<f[n];
    return 0;
}