[BZOJ1096][ZJOI2007]仓库建设（斜率优化dp）

仿佛终于理解了斜率优化

Address

https://www.lydsy.com/JudgeOnline/problem.php?id=1096

Solution

显然 dp 。
定义状态：
$f[i]$ 表示在前 $i$ 个工厂建立仓库，且第 $i$ 个工厂必须建立仓库的最小费用。
转移就是枚举上一个仓库的建立位置 $j$ （如果没有则 $j=0$ ）

$$f[i]=\min_{j=0}^{i-1}\{f[j]+\sum_{k=j+1}^i\{(X_i-X_k)\times P_k\}+C_i\}$$
考虑把 $\sum_{k=j+1}^i\{(X_i-X_k)\times P_k\}$ 化开：
（ $S$ 为 $P$ 的前缀和， $R$ 为 $P\times X$ 的前缀和）
（即 $S_i=\sum_{j=1}^iP_j$ ， $R_i=\sum_{j=1}^i\{P_j\times X_j\}$ ）
$$\sum_{k=j+1}^i\{(X_i-X_k)\times P_k\}=\sum_{k=j+1}^i\{(X_i\times P_k-X_k\times P_k\}$$
$$=\sum_{k=j+1}^i\{X_i\times P_k\}-\sum_{k=j+1}^i\{X_k\times P_k\}$$
$$=X_i\times\sum_{k=j+1}^iP_k-\sum_{k=j+1}^i\{X_k\times P_k\}$$
$$=X_i\times(S_i-S_j)-(R_i-R_j)$$
于是：
$$f[i]=\min_{j=0}^{i-1}\{f[j]+X_i\times(S_i-S_j)-(R_i-R_j)+C_i\}$$
$$=\min_{j=0}^{i-1}\{f[j]+X_i\times S_i-X_i\times S_j-R_i+R_j+C_i\}$$
$$=\min_{j=0}^{i-1}\{f[j]+R_j-X_i\times S_j\}+X_i\times S_i-R_i+C_i$$
这是一个显然的斜率优化。
设 $k<j<i$ ，那么从 $j$ 转移比 $k$ 优当且仅当：
$$f[j]+R_j-X_i\times S_j<f[k]+R_k-X_i\times S_k$$
$$(f[j]+R_j)-(f[k]+R_k)<X_i\times S_j-X_i\times S_k$$
$$\frac{(f[j]+R_j)-(f[k]+R_k)}{S_j-S_k}<X_i$$
左边是一个斜率的式子！
把每个决策看作一个平面上一个点：决策为 $i$ 为 $(S_i,f[i]+R_i)$ 。
考虑 $a<b<c$ ，设 $slope[a,b]$ 为过第 $a$ 个点与第 $b$ 个点的直线的斜率，
并且 $slope[a,b]>slope[b,c]$ 。
（1）如果 $slope[a,b]<X_i$ ，那么 $slope[b,c]$ 一定也小于 $X_i$。这时候决策 $c$ 比决策 $b$ 优。
（2）如果 $slope[b,c]>X_i$ ，那么 $slope[a,b]$ 一定也大于 $X_i$ 。那么这时候决策 $b$ 一定不比决策 $a$ 优。
（3）如果 $slope[b,c]<X_i<slope[a,b]$ ，那么决策 $b$ 一定不比决策 $a$ 和 $c$ 优。
得出结论：如果 $slope[a,b]>slope[b,c]$ ，那么决策 $b$ 一定不会是最优决策。
所以，所有可能成为最优决策的点构成一个下凸壳，斜率单调递增。
并且，如果一个决策 $j$ 在下凸壳上是最优决策（设 $k$ 为 $j$ 在下凸壳上的前一个点， $h$ 为 $j$ 在凸壳上的后一个点），那么一定有 $slope[k,j]<X_i,slope[j,h]>X_i$ 。从这里也能得出，这个动规方程在下凸壳上满足决策单调性。
根据斜率的单调性，可以用一个单调队列维护下凸壳，每次进行决策时，利用决策的单调性，从队首不断出队，找到最优决策 $j$ 并转移。每次决策结束后，在下凸壳末尾加入点 $(S_i,f[i]+R_i)$ ，并维护下凸的性质。
由于每个点最多进出队一次，所以复杂度 $O(n)$ 。

Code

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define For(i, a, b) for (i = a; i <= b; i++)
using namespace std;
inline int read() {
    int res = 0; bool bo = 0; char c;
    while (((c = getchar()) < '0' || c > '9') && c != '-');
    if (c == '-') bo = 1; else res = c - 48;
    while ((c = getchar()) >= '0' && c <= '9')
        res = (res << 3) + (res << 1) + (c - 48);
    return bo ? ~res + 1 : res;
}
typedef long long ll; const int N = 1e6 + 5;
int n, X[N], P[N], C[N], H, T, que[N]; ll f[N], S[N], R[N], x[N], y[N];
ll calc(int j, int i) {return y[j] - x[j] * X[i];}
double slope(int j, int k) {
    return 1.0 * (y[j] - y[k]) / (x[j] - x[k]);
}
int main() {
    int i; n = read(); For (i, 1, n) X[i] = read(), P[i] = read(), C[i] = read();
    For (i, 1, n) S[i] = S[i - 1] + P[i], R[i] = R[i - 1] + 1ll * P[i] * X[i];
    f[0] = x[0] = y[0] = 0; que[H = T = 1] = 0; For (i, 1, n) {
        while (H < T && calc(que[H], i) >= calc(que[H + 1], i)) H++;
        f[i] = calc(que[H], i) + 1ll * S[i] * X[i] - R[i] + C[i];
        x[i] = S[i]; y[i] = f[i] + R[i];
        while (H < T && slope(que[T - 1], que[T]) > slope(que[T], i)) T--;
        que[++T] = i;
    }
    cout << f[n] << endl; return 0;
}