【洛谷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] imes (sum[i]-sum[k]))(0<k<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('
'), 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] imes (sum[i]-sum[j]))(0<j<i) ]

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

[g[j]+sum[j] imes (sum[i]-sum[j])>g[k]+sum[k] imes (sum[i]-sum[k]) ]

进行一些变换,得到:

[g[j]-sum[j]^2>g[k]-sum[k]^2+sum[i] imes (-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'](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('
'), write(st);
		for (int i = k; i > 1; i--)
			putchar(' '), write(st = pre[i][st]);
		return 0;
	}
}
int main()
{
	return zyt::work();
}
原文地址:https://www.cnblogs.com/zyt1253679098/p/9920465.html