【洛谷3648/BZOJ3675】[APIO2014]序列分割（斜率优化DP）

题目：

洛谷3648

注：这道题洛谷3648有SPJ，要求输出方案。BZOJ3675数据组数较多但不要求输出方案。

分析：

这可能是我第三次重学斜率优化了……好菜啊

这道题首先一看就是个DP。稍微推一推类似下面这种式子就会发现事实上结果和切的顺序无关

$$a(b+c)+bc=ab+c(a+b)=ab+ac+bc$$

那么就可以用$f[i][j]$表示切了$j$次，最右一次在$i$后面切的最大值。用$sum[i]$表示原序列前$i$个数之和，那么就有了这个DP方程（假设在$i$后面是第一次切割）：

$$f[i][j]=max(f[k][j-1]+sum[k]\times (sum[i]-sum[k]))(0<k<i)$$

其中$j$这一维可以滚动存储。转移的时候顺便记一下从什么地方转移过来就行。这个时间复杂度是$O(n^{2}k)$。丢一部分代码。

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]$。再来看这个式子。

$$f[i]=max(g[j]+sum[j]\times (sum[i]-sum[j]))(0<j<i)$$

考虑对于$g[j]$和$g[k](j<k<i)$，如果从$g[j]$转移到$f[i]$比从$g[k]$转移更优，那么一定满足：

$$g[j]+sum[j]\times (sum[i]-sum[j])>g[k]+sum[k]\times (sum[i]-sum[k])$$

进行一些变换，得到：

$$g[j]-sum[j{]}^2>g[k]-sum[k{]}^2+sum[i]\times (-sum[j]+sum[k])$$

由于前缀和单调不降，$(-sum[j]+sum[k])$是非负的，除到右边得到（暂时不考虑为$-sum[j]+sum[k]=0$的特殊情况）：

$$\frac{(g[j]-sum[j{]}^2)-(g[k]-sum[k{]}^2)}{-sum[j]+sum[k]}>sum[i]$$

可以看出左侧的式子很“工整”。把$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^{\prime }](i<i^{\prime }\le 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();
}