【BZOJ4518】【SDOI2016】征途（斜率优化DP）

Description

Pine开始了从S地到T地的征途。
从S地到T地的路可以划分成n段，相邻两段路的分界点设有休息站。
Pine计划用m天到达T地。除第m天外，每一天晚上Pine都必须在休息站过夜。所以，一段路必须在同一天中走完。
Pine希望每一天走的路长度尽可能相近，所以他希望每一天走的路的长度的方差尽可能小。
帮助Pine求出最小方差是多少。
设方差是v，可以证明，v×m^2是一个整数。为了避免精度误差，输出结果时输出v×m^2。

---

Solution

我是来复习斜率优化板子的。。。
这题其实是要求$\displaystyle\sum x_i^2 - Sum^2$的最小值。
然后列出转移方程后，颓一下式子，斜率优化就行了。。

---

Code

/************************************************
 * Au: Hany01
 * Date: Apr 9th, 2018
 * Prob: [BZOJ4518][SDOI2016] 征途
 * Email: hany01@foxmail.com
************************************************/

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;
#define File(a) freopen(a".in", "r", stdin), freopen(a".out", "w", stdout)
#define rep(i, j) for (register int i = 0, i##_end_ = (j); i < i##_end_; ++ i)
#define For(i, j, k) for (register int i = (j), i##_end_ = (k); i <= i##_end_; ++ i)
#define Fordown(i, j, k) for (register int i = (j), i##_end_ = (k); i >= i##_end_; -- i)
#define Set(a, b) memset(a, b, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define x first
#define y second
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define ALL(a) (a).begin(), (a).end()
#define SZ(a) ((int)(a).size())
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define Mod (1000000007)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define y1 wozenmezhemecaia

template <typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template <typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }

inline int read()
{
    register int _, __; register char c_;
    for (_ = 0, __ = 1, c_ = getchar(); c_ < '0' || c_ > '9'; c_ = getchar()) if (c_ == '-') __ = -1;
    for ( ; c_ >= '0' && c_ <= '9'; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
    return _ * __;
}

const int maxn = 3005;

int n, m, sum[maxn], a[maxn];
LL f[maxn], g[maxn];

inline double Slope(int j, int k) {
    return ((g[j] + sum[j] * sum[j]) - (g[k] + sum[k] * sum[k])) * 1. / (sum[j] - sum[k]);
}

inline void DP(int cur)
{
    static int q[maxn], head, tail;
    head = 1, tail = 0;
    For(i, cur, n) {
        q[++ tail] = i - 1;
        while (head + 1 < tail && Slope(q[tail - 2], q[tail - 1]) >= Slope(q[tail - 1], q[tail])) q[tail - 1] = q[tail], -- tail;
        while (head < tail && Slope(q[head], q[head + 1]) <= 2 * sum[i]) ++ head;
        register int j = q[head];
        f[i] = g[j] + (sum[i] - sum[j]) * (sum[i] - sum[j]);
    }
}

int main()
{
#ifdef hany01
    File("bzoj4518");
#endif

    n = read(), m = read();
    For(i, 1, n) sum[i] = sum[i - 1] + (a[i] = read());

    For(i, 1, n) g[i] = INF;
    For(i, 1, m) DP(i), swap(f, g);

    printf("%lld\n", g[n] * (LL)m - sum[n] * sum[n]);

    return 0;
}
//Error(s):
//最后输出的是g而不是f
//g没有赋最大值

//秋风起兮白云飞，草木黄落兮雁南归。
//    -- 刘彻《秋风辞》