【BZOJ3675】【APIO2014】序列分割（斜率优化DP）

Description

小H最近迷上了一个分隔序列的游戏。在这个游戏里，小H需要将一个长度为n的非负整数序列分割成k+1个非空的子序列。为了得到k+1个子序列，小H需要重复k次以下的步骤：
1.小H首先选择一个长度超过1的序列（一开始小H只有一个长度为n的序列——也就是一开始得到的整个序列）；
2.选择一个位置，并通过这个位置将这个序列分割成连续的两个非空的新序列。
每次进行上述步骤之后，小H将会得到一定的分数。这个分数为两个新序列中元素和的乘积。小H希望选择一种最佳的分割方式，使得k轮之后，小H的总得分最大。

Solution

刷了一下午、晚上的斜率优化，终于tmd有一个不是模板题的了。。。
手玩可以发现得分即：

$$\prod_{i=1}^{k+1}\sum_{j=l_i}^{r_i}a_j\sum_{j=1}^{l_i-1}a_j$$
$dp_{i,j}$表示前 $i$个分了$k$次，转移非常显然吧。
然后你会发现空间不够用，于是搞成滚动的。
然后你会发现时间更不够用，于是搞一个斜率优化，时间复杂度 $O(nk)$。
然而 hzwer神犇用单调队列也A掉了，%%%
这篇博客也讲得挺好的。

Code

/**************************
Au: Hany01
Date: Jan 7th, 2018
Prob: bzoj3675 & apio 2014
Email: hany01@foxmail.com
**************************/

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;
#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 fir first
#define sec 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 Ha (1000000007)

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 _ * __;
}

inline void File()
{
#ifdef hany01
    freopen("bzoj3675.in", "r", stdin);
    freopen("bzoj3675.out", "w", stdout);
#endif
}

const int maxn = 100005;

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

inline void Init()
{
    n = read(); k = read();
    for (register int i = 1; i <= n; ++ i)
        if (!(a[i] = read())) -- i, -- n;
    For(i, 1, n) sum[i] = sum[i - 1] + a[i];
}

#define x(i) sum[i]
#define y(i) (f[i] - sum[i] * sum[i])
inline double slope(int k, int j) { return (y(j) - y(k)) * 1.0 / (x(j) - x(k)); }

inline void DP(int cur)
{
    register int j, 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]) >= -sum[i]) ++ head;
        j = q[head]; g[i] = f[j] + sum[j] * (sum[i] - sum[j]);
    }
}

inline void Solve()
{
    For(i, 1, k) {
        DP(i);
        For(j, i, n) swap(f[j], g[j]);
    }
    printf("%lld\n", f[n]);
}

int main()
{
    File();
    Init();
    Solve();
    return 0;
}
//水风轻，蘋花渐老，月露冷、梧叶飘黄。
//    -- 柳永《玉蝴蝶·望处雨收云断》