洛谷 P3620 [APIO/CTSC 2007]数据备份（wqs二分优化 dp）

最新推荐文章于 2025-08-26 17:41:51 发布

原创最新推荐文章于 2025-08-26 17:41:51 发布 · 300 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#wqs二分 #wqs二分优化dp

本文探讨了在特定约束条件下，如何通过动态规划（DP）寻找最优解，并引入了一种有效的优化策略。通过二分搜索确定最佳权值，实现对DP状态转移的优化，从而在O(n log n)的时间复杂度内解决问题。文章详细解释了优化过程，并提供了完整的代码实现。

在这里插入图片描述

最优解肯定是选择了 k 对相邻的办公楼，考虑直接 dp：dp[i][j] 表示前 i 栋楼选了 j 对的最小花费，转移方程：dp[i][j] = min(dp[i - 1][j],dp[i - 1][j - 1] + a[i] - a[j])

复杂度为 $O(n^2)$ 。考虑优化：若不限制选几对， $i, d p [n] [i]$ 在二维平面上肯定是一个下凸函数，二分一个权值 $s l o p e$ ，使得每个点的答案要减去 $s l o p e * i$ ，即答案 = $d p [n] [i] - s l o p e * i$ ，从线性规划的角度考虑，这相当于二分一个斜率 $s l o p e$ ，要让这条直线在这个凸包上切某个点截距最小，由于凸包斜率具有单调性，因此可以二分 $s l o p e$ 。

也可以更直接的理解：若没有限制，最小答案肯定是一条电缆也不用。为每对电缆扣去一个权值 $s l o p e$ ，然后在不考虑限制的情况下进行 dp 求解， $s l o p e$ 越大，则最优解使用的电缆越多，否则最优解使用的电缆越少，具有单调性。

二分 $s l o p e$ ，找到最优时，电缆数 = k的答案。有可能二分不到这个电缆数 = k 的情况，考虑转移时，每个状态取得最优的情况下尽可能的多使用电缆。二分时找到第一个满足使用电缆数 $≥k\geq k$ 的解。

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
typedef long long ll;
const ll inf = 1e14;
int n,k,a[maxn];
ll dp[maxn],tp[maxn],sum;
ll solve(ll x) {
	tp[0] = dp[0] = 0;
	for (int i = 1; i <= n; i++) dp[i] = sum, tp[i] = 0;;
	dp[1] = 0;
	for (int i = 2; i <= n; i++) {
		dp[i] = dp[i - 1]; tp[i] = tp[i - 1];
		if (dp[i - 2] + a[i] - a[i - 1] + x < dp[i]) {
			dp[i] = dp[i - 2] + x + a[i] - a[i - 1];
			tp[i] = tp[i - 2] + 1;
		} else if (dp[i - 2] + x + a[i] - a[i - 1] == dp[i]) {
			if (tp[i - 2] + 1 > tp[i])
				tp[i] = tp[i - 2] + 1;
		}
	}
	return tp[n];
}
int main() {
	scanf("%d%d",&n,&k);
	for (int i = 1; i <= n; i++)
		scanf("%d",&a[i]), sum += a[i];
	ll l = -sum, r = 0;
	while (l < r) {
		ll mid = l + r >> 1;
		if (solve(mid) < k) r = mid;
		else l = mid + 1;
	}
	solve(l - 1);
	printf("%lld\n",dp[n] - 1ll * k * (l - 1));
	return 0;
}