2018.11.02【HNOI2008】【BZOJ1010】【洛谷P3195】玩具装箱（斜率优化DP）

BZOJ传送门

洛谷传送门

---

解析：

看到平方项多半就是两种套路，决策单调性和斜率优化，这道题斜率优化可以$O(n)$。

首先还是推DP式子，这个很好想（$sum$表示前缀和）。 f i = min ⁡ j = 1 i − 1 { ( i − j + s u m i − s u m j − L − 1 ) 2 + f j } f_i=\min_{j=1}^{i-1}\{(i-j+sum_i-sum_j-L-1)^2+f_j\} fi​=j=1mini−1​{(i−j+sumi​−sumj​−L−1)2+fj​}

是在是太长了，代换一下，令$A(i)=su{m}_i+i,B(i)=A(i)+L+1$，则状态转移方程化简为 f i = min ⁡ j = 1 i − 1 { ( A ( i ) − B ( j ) ) 2 + f j } f_i=\min_{j=1}^{i-1}\{(A(i)-B(j))^2+f_j\} fi​=j=1mini−1​{(A(i)−B(j))2+fj​} = A ( i ) 2 + min ⁡ j = 1 i − 1 { − 2 A ( i ) B ( j ) + B ( j ) 2 + f j } =A(i)^2+\min_{j=1}^{i-1}\{-2A(i)B(j)+B(j)^2+f_j\} =A(i)2+j=1mini−1​{−2A(i)B(j)+B(j)2+fj​}

然后每个决策点设置为$(B(j),B(j{)}^2+f_j)$，可以斜率优化了。

发现决策的斜率是单调的，单调队列维护一下凸包就行了。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define pc putchar
#define cs const

cs int N=50004;
int q[N];
int head=1,tail=1;
ll n,L;
ll sum[N],f[N];

inline ll A(int i){return sum[i]+i;}
inline ll B(int i){return sum[i]+i+L+1;}
inline ll calc(int i,int j){
    return f[j]+(A(i)-B(j))*(A(i)-B(j));
}
inline ll X(int i){return B(i);}
inline ll Y(int i){return f[i]+B(i)*B(i);}
inline double slope(int i,int j){
    return (1.0*Y(i)-Y(j))/(1.0*X(i)-X(j));
}

signed main(){
    scanf("%lld%lld",&n,&L);
    for(int re i=1;i<=n;++i)scanf("%lld",&sum[i]),sum[i]+=sum[i-1];
    for(int re i=1;i<=n;++i){
        while(head<tail&&calc(i,q[head])>calc(i,q[head+1]))++head;
        f[i]=calc(i,q[head]);
        while(head<tail&&slope(i,q[tail-1])<slope(q[tail],q[tail-1]))--tail;
        q[++tail]=i;
    }
    printf("%lld",f[n]);
    return 0;
}