[BZOJ1010][HNOI2008]玩具装箱toy-斜率优化

玩具装箱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。但他希望费用最小.

Input

　　第一行输入两个整数N，L.接下来N行输入Ci.1<=N<=50000,1<=L,Ci<=10^7

Output

　　输出最小费用

Sample Input

5 4
3
4
2
1
4

Sample Output

1

---

斜率优化的入门题……
然而当时却没有去写……
然后就因为斜率优化写得不够熟而被同学出的题给坑了……

(╯‵□′)╯︵┻━┻

---

思路:
可以很快想到一个很显然的$O(n^2)$的dp：
$dp[i]=max\{dp[j]+(c[i]+i-c[j]-j-1-l)^2\}$

然后咱试图去优化它。
首先，发现c数组的值总是和对应的i共存，那么咱令$f[i]=c[i]+i$，$L=l+1$，
那么$dp[i]=max\{dp[j]+(f[i]-f[j]+L)^2\}$。

通过展开可以证明上式具有单调性。
（其实有一个通用的技巧是，不去证单调性而是写好后和暴力对拍观察有无单调性）
于是考虑斜率优化:
设在位置i，k的决策比j优。
那么$dp[k]+(f[i]-f[k]+L)^2 \gt dp[j]+(f[i]-f[j]+L)^2$
把$f[i]+L$视作一项，对上式进行展开并化简得:
$dp[k]-2*(f[i]+L)*f[k]+f[k]^2 \gt dp[j]-2*(f[i]+L)*f[j]+f[j]^2$

整理得：
$dp[j]-dp[k]-2*L*(f[j]-f[k])+f[j]^2-f[k]^2 \lt 2*f[i]*(f[j]-f[k])$

给左半边配个方:
$dp[j]-dp[k]+(f[j]+L)^2-(f[k]+L)^2 \lt 2*f[i]*(f[j]-f[k])$

把它写成斜率的形式:
$\frac {dp[j]-dp[k]+(f[j]+L)^2-(f[k]+L)^2}{2*(f[j]-f[k])} \lt f[i]$

于是咱得到了一个可以$O(1)$判断最优解的式子，咱称他为$slope(i,j)$。

当j < k且上式被满足，则k比j更优。
由于决策单调性，可以维护一个单调队列，做到$O(1)$转移:
当$slope(q[l],q[l+1]) \lt f[i]$，$q[l+1]$更优，弹出$q[l]$；
当$slope(q[r-1],q[r]) \gt slope(q[r],i)$，则留下的$q[r]$必会在$q[r-1]$被弹出的同时弹出，为无效状态，于是弹出$q[r]$。
每次从队首获取最优决策即可~

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>

using namespace std;

typedef long long ll;
typedef double db;

inline long long read()
{
    long long x=0;char ch=getchar();
    while(ch<'0' || '9'<ch)ch=getchar();
    while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
    return x;
}

const int N=50009;

ll n,l,p;
ll dp[N],f[N],c[N];
ll q[N],ls,r;

inline db sqr(ll a)
{
    return (db)a*a;
}

inline db slope(int j,int k)
{
    return ((db)dp[k]-dp[j]+sqr(f[k]+l)-sqr(f[j]+l))/(2.0*(f[k]-f[j]));
}

int main()
{
    n=read();
    l=read()+1;

    for(int i=1;i<=n;i++)
    {
        c[i]=read()+c[i-1];
        f[i]=c[i]+i;
    }

    ls=1,r=0;
    q[++r]=0;
    for(int i=1;i<=n;i++)
    {
        while(ls<r && slope(q[ls],q[ls+1])<=f[i])
            ls++;
        dp[i]=dp[q[ls]]+(ll)sqr(f[i]-f[q[ls]]-l);
        while(ls<r && slope(q[r],i)<slope(q[r-1],q[r]))
            r--;
        q[++r]=i;
    }

    printf("%lld\n",dp[n]);
    return 0;
}