HDU 3507 Print Article——斜率优化dp

首先不难推出这道题的状态转移方程：dp[i]=min{dp[j]+(sum[i]-sum[j])^2|0<j<i}

假设有k<j,j比k更优当且仅当dp[j]+(sum[i]-sum[j])^2<=dp[k]+(sum[i]-sum[k])^2

化简得：dp[j]-dp[k]+sum[j]^2-sum[k]^2<=2*sum[i]*(sum[j]-sum[k])

设f[i]=dp[i]+sum[i]^2

则(f[j]-f[k])/(sum[j]-sum[k]) <= 2*sum[i]

将(f,sum)看做二维坐标系上的点，那么上面等式的左边可以看做一个斜率

设k[j, k] = (f[j], f[k])/(sum[j]-sum[k])，则k[j,k]代表k指向j的线的斜率，根据上面的推到，g[j,k]<=2*sum[i]代表j比k优，否则k比j优

可以证明上凸的图像一定不优，所以我们要维护下凸的图像（下面会证明）

那么当前的最优点实际上是斜率小于2*sum[i]的最靠后的线对应的尾端点

 

上凸图像一定不优的证明：

假设k<j<i构成一个上凸图像，则g(j, k)>g(i, j);

此时他们和2*sum[i]有三种关系：

1.g(j, k)>g(i,j)>2*sum[i],此时k比j优

2.g(j, k)>2*sum[i]>g(i,j),此时i比j优

3.2*sum[i]>g(j, k)>g(i,j),此时i和k都比j优

所以j一定不优

 

我们发现2*sum[i]是单调递增的，所以可以用单调队列来维护，当然如果有的题目没有单调递增的性质我们可以二分

然后发现莫名的好写。。。注意long long

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 500010;
int N, M, q[maxn], l, r;
LL sum[maxn], dp[maxn];
inline LL f(int i) { return dp[i] + sum[i]*sum[i]; }
int main() {
    while (~scanf("%d%d", &N, &M)) {
        for (int i = 1; i <= N; i++) scanf("%lld", &sum[i]), sum[i] += sum[i-1];
        l = 0, r = 0, q[0] = 0;
        for (int i = 1; i <= N; i++) {
            while (l < r && f(q[l+1])-f(q[l]) <= 2LL*sum[i]*(sum[q[l+1]]-sum[q[l]])) l++;
            dp[i] = dp[q[l]]+(sum[i]-sum[q[l]])*(sum[i]-sum[q[l]])+M;
            while (l < r && (f(q[r])-f(q[r-1]))*(sum[i]-sum[q[r]]) >= (f(i)-f(q[r]))*(sum[q[r]]-sum[q[r-1]])) r--;
            q[++r] = i;
        }
        printf("%lld\n", dp[N]);
    }
    return 0;
}