bzoj1911 [Apio2010]特别行动队

Description

---

你有一支由 n 名预备役士兵组成的部队，士兵从 1 到 n 编号，要将他们拆分 成若干特别行动队调入战场。出于默契的考虑，同一支特别行动队中队员的编号 应该连续，即为形如(i,i+1,…,i+k)(i, i + 1, …, i + k)(i,i+1,…,i+k) 的序列。 编号为 i 的士兵的初始战斗力为 xi ，一支特别行动队的初始战斗力 x 为队内 士兵初始战斗力之和，即 x=xi+xi+1+…+xi+kx = x_i + x_{i+1} + … + x_{i+k}x=xi​+xi+1​+…+xi+k​ 。
通过长期的观察，你总结出一支特别行动队的初始战斗力 x 将按如下经验公 式修正为 x′:x′=ax2+bx+cx’:x’= ax^2+bx+cx′:x′=ax2+bx+c ，其中 a, b, c 是已知的系数（a < 0）。 作为部队统帅，现在你要为这支部队进行编队，使得所有特别行动队修正后 战斗力之和最大。试求出这个最大和。
例如，你有 4 名士兵， x1=2,x2=2,x3=3,x4=4x_1 = 2, x_2 = 2, x_3 = 3, x_4 = 4x1​=2,x2​=2,x3​=3,x4​=4 。经验公式中的参数为 a = –1, b = 10, c = –20。此时，最佳方案是将士兵组成 3 个特别行动队：第一队包含士兵 1 和士兵 2，第二队包含士兵 3，第三队包含士兵 4。特别行动队的初始战斗力分 别为 4, 3, 4，修正后的战斗力分别为 4, 1, 4。修正后的战斗力和为 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

Solution

---

月考凉凉，只好潜心做题

十分naive的斜率优化，相信本蒟蒻不写过程各位姥爷们也能意会到精髓吧

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define sqr(x) (((LL)x)*((LL)x))

typedef long long LL;
typedef double db;
const int N=2000005;

LL f[N],s[N],a[N];
LL A,B,C;
int queue[N];

db slope(int k,int j) {
    db ret=f[j]-f[k]+(db)A*(sqr(s[j])-sqr(s[k]))+B*(s[k]-s[j]);
    ret=ret/(2.0*A*(s[j]-s[k]));
    return ret;
}

LL get_f(LL x) {
    return A*sqr(x)+B*x+C;
}

void solve(int n) {
    int head=1,tail=0;
    queue[++tail]=0;
    rep(i,1,n) {
        while (head<tail&&slope(queue[head],queue[head+1])<=s[i]) head++;
        f[i]=f[queue[head]]+get_f(s[i]-s[queue[head]]);
        while (head<tail&&slope(queue[tail-1],queue[tail])>slope(queue[tail],i)) tail--;
        queue[++tail]=i;
    }
    printf("%lld\n", f[n]);
}

int main(void) {
    int n; scanf("%lld%lld%lld%lld",&n,&A,&B,&C);
    rep(i,1,n) scanf("%lld",&a[i]);
    rep(i,1,n) s[i]=s[i-1]+a[i];
    solve(n);
    return 0;
}