bzoj1010 [HNOI2008]玩具装箱toy

Description

---

P教授要去看奥运，但是他舍不下他的玩具，于是他决定把所有的玩具运到北京。他使用自己的压缩器进行压
缩，其可以将任意物品变成一堆，再放到一种特殊的一维容器中。P教授有编号为1…N的N件玩具，第i件玩具经过
压缩后变成一维长度为Ci.为了方便整理，P教授要求在一个一维容器中的玩具编号是连续的。同时如果一个一维容
器中有多个玩具，那么两件玩具之间要加入一个单位长度的填充物，形式地说如果将第i件玩具到第j个玩具放到一
个容器中，那么容器的长度将为 x=j-i+Sigma(Ck) i<=K<=j 制作容器的费用与容器的长度有关，根据教授研究，
如果容器长度为x,其制作费用为(X-L)^2.其中L是一个常量。P教授不关心容器的数目，他可以制作出任意长度的容
器，甚至超过L。但他希望费用最小.

1<=N<=50000,1<=L,Ci<=10^7

Solution

---

设f[i]表示前i个物品的最优解，方程就能写成
$f[i]=min\{f[j]+(i-j-1+sum[i]-sum[j]-L)^2\}$，其中sum[i]表示前缀和
方便起见令$g[i]=sum[i]+i，c=L+1$
那么$f[i]=min\{f[j]+(g[i]-g[j]-c)^2\}$
我们任取三个点i、j、k，且k< j< i，对于i而言从j转移更优的情况当且仅当满足
$f[j]+(g[i]-g[j]-c)^2\leq f[k]+(g[i]-g[k]-c)^2$
随意化简一下得到$\frac{f[j]+(g[j]+c)^2-f[k]-(g[k]+c)^2}{2(g[j]-g[k])}\leq g[i]$
注意这里分母的正负号，i、j、k的大小顺序，以此判断一下斜率是递增还是递减的，然后就能根据套路上单调队列惹

感觉今天才对这类题有了比较清晰的认识，看来还是题做得不够

Code

---

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

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

int a[N],c;
int queue[N];
LL f[N],g[N];

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

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])<=g[i]) head++;
        f[i]=f[queue[head]]+sqr(g[i]-g[queue[head]]-c);
        while (head<tail&&slope(queue[tail-1],queue[tail])>slope(queue[tail],i)) tail--;
        queue[++tail]=i;
    }
}

int main(void) {
    int n; scanf("%d%d",&n,&c); c++;
    rep(i,1,n) scanf("%d",&a[i]);
    rep(i,1,n) g[i]=g[i-1]+a[i];
    rep(i,1,n) g[i]=g[i]+i;
    solve(n);
    printf("%lld\n", f[n]);
    return 0;
}