【洛谷3648/BZOJ3675】[APIO2014]序列分割（斜率优化DP）
题目：
注：这道题洛谷3648有SPJ，要求输出方案。BZOJ3675数据组数较多但不要求输出方案。
分析：
这可能是我第三次重学斜率优化了……好菜啊
这道题首先一看就是个DP。稍微推一推类似下面这种式子就会发现事实上结果和切的顺序无关
那么就可以用\(f[i][j]\)表示切了\(j\)次，最右一次在\(i\)后面切的最大值。用\(sum[i]\)表示原序列前\(i\)个数之和，那么就有了这个DP方程（假设在\(i\)后面是第一次切割）：
其中\(j\)这一维可以滚动存储。转移的时候顺便记一下从什么地方转移过来就行。这个时间复杂度是\(O(n^2k)\)。丢一部分代码。
int work()
{
read(n), read(k);
for (int i = 1; i <= n; i++)
read(sum[i]), sum[i] += sum[i - 1];
for (int i = 2; i <= k; i++)
for (int j = n; j > 0; j--)
for (int g = 1; g < j; g++)
{
if (dp[j] < dp[g] + sum[g] * (sum[j] - sum[g]))
{
dp[j] = dp[g] + sum[g] * (sum[j] - sum[g]);
pre[i][j] = g;
}
}
int st = 0;
ll ans = 0;
for (int i = 1; i <= n; i++)
if (ans < dp[i] + sum[i] * (sum[n] - sum[i]))
{
ans = dp[i] + sum[i] * (sum[n] - sum[i]);
st = i;
}
write(ans), putchar('\n'), write(st);
for (int i = k; i > 1; i--)
putchar(' '), write(st = pre[i][st]);
return 0;
}
实测洛谷上有\(64\)分。64分够了，本文结束。
考虑斜率优化。以下的讨论均为在\(j\)（切的次数）固定的情况下，为方便说明，用\(f[i]\)表示\(f[i][j]\)，\(g[i]\)表示\(f[i][j-1]\)。再来看这个式子。
考虑对于\(g[j]\)和\(g[k](j<k<i)\)，如果从\(g[j]\)转移到\(f[i]\)比从\(g[k]\)转移更优，那么一定满足：
进行一些变换，得到：
由于前缀和单调不降，\((-sum[j]+sum[k])\)是非负的，除到右边得到（暂时不考虑为\(-sum[j]+sum[k]=0\)的特殊情况）：
可以看出左侧的式子很“工整”。把\(g[j]-sum[j]^2\)看作点\(j\)的纵坐标，\(-sum[j]\)看作点\(j\)的横坐标，则左侧就是\(j\)和\(k\)两点之间的斜率。对于横坐标相等的两点，斜率根据纵坐标的符号视作正无穷或负无穷。
接下来阅读前，请时刻牢记：对于\(j<k<i\)，如果\(j\)和\(k\)之间的斜率大于\(sum[i]\)，则\(j\)比\(k\)优
同时还有这句话的反面：对于\(j<k<i\)，如果\(j\)和\(k\)之间的斜率不大于\(sum[i]\)，则\(j\)比\(k\)劣
由于\(sum[i]\)是单调不降的，所以满足决策单调性，即：如果对于\(f[i]\)，从\(g[k]\)转移比从\(g[j]\)优\((j<k)\)，则对于\(f[i'](i<i'\leq n)\)，\(g[j]\)不可能是最优的（显然，\(j\)和\(k\)间的斜率是不受\(i\)影响的，而如果此时斜率已经小于等于\(sum[i]\)了，则以后也不可能大于\(sum[i]\)，所以\(j\)以后永远不可能比\(k\)优） 。
那么可以维护一个斜率递增的单调队列。如果感到这部分难以理解，请再反复看上面三句加粗的话。当决策\(f[i]\)时，弹出单调队列首的若干元素，直到只剩一个元素或前两个元素的斜率大于\(sum[i]\)。当尝试插入\(i\)点时，弹出队尾的若干元素，直到只剩一个元素或队尾与\(i\)的斜率比队尾后两个元素大。在几何意义上，这是一个下凸包。读者可以画图理解。
代码：
#include <cstdio>
#include <algorithm>
#include <cctype>
#include <cstring>
using namespace std;
namespace zyt
{
template<typename T>
inline void read(T &x)
{
bool f = false;
char c;
x = 0;
do
c = getchar();
while (c != '-' && !isdigit(c));
if (c == '-')
f = true, c = getchar();
do
x = x * 10 + c - '0', c = getchar();
while (isdigit(c));
if (f)
x = -x;
}
template<typename T>
inline void write(T x)
{
static char buf[20];
char *pos = buf;
if (x < 0)
putchar('-'), x = -x;
do
*pos++ = x % 10 + '0';
while (x /= 10);
while (pos > buf)
putchar(*--pos);
}
typedef long long ll;
typedef long double ld;
const int N = 1e5 + 10, K = 210;
const ll LINF = 0x3f3f3f3f3f3f3f3fLL;
int n, k, now;
ll sum[N], dp[2][N];
int pre[K][N];
inline ll sq(const ll x)
{
return x * x;
}
inline ll y(const int i)
{
return dp[now ^ 1][i] - sq(sum[i]);
}
inline ll x(const int i)
{
return -sum[i];
}
inline ld ratio(const int i, const int j)
{
if (x(i) == x(j))
return (y(i) - y(j) > 0) ? LINF : -LINF;
else
return (ld)(y(i) - y(j)) / (x(i) - x(j));
}
int work()
{
static int q[N];
read(n), read(k);
for (int i = 1; i <= n; i++)
read(sum[i]), sum[i] += sum[i - 1];
for (int i = 2; i <= k; i++)
{
now = i & 1;
int h = 0, t = 1;
q[0] = 0;
for (int j = 1; j <= n; j++)
{
while (h + 1 < t && ratio(q[h], q[h + 1]) <= sum[j])
++h;
dp[now][j] = dp[now ^ 1][q[h]] + sum[q[h]] * (sum[j] - sum[q[h]]);
pre[i][j] = q[h];
while (h + 1 < t && ratio(q[t - 2], q[t - 1]) >= ratio(q[t - 1], j))
--t;
q[t++] = j;
}
}
int st = 0;
ll ans = 0;
for (int i = 1; i <= n; i++)
if (ans < dp[now][i] + sum[i] * (sum[n] - sum[i]))
{
ans = dp[now][i] + sum[i] * (sum[n] - sum[i]);
st = i;
}
write(ans), putchar('\n'), write(st);
for (int i = k; i > 1; i--)
putchar(' '), write(st = pre[i][st]);
return 0;
}
}
int main()
{
return zyt::work();
}
