二分+DP The 2023 ICPC Asia Hefei Regional Contest Problem G Streak Manipulation

题意

给定一个长度为$n$的01串，最多可以将$m$个0变成1，要求使01串中第$k$长连续1串的长度最大。
输出最大值，如果无法得到第$k$长的连续1串，输出-1

做法

考虑二分答案，二分第$k$长连续1串的长度。
用DP来check
对于当前的$tar$, 设$dp[i][j][0/1]$为前$i$个字符，有$j$段长度大于等于$tar$的连续1串，且当前字符是否位于第$j$段连续1串的末尾（0为假1为真）所需的最少操作次数。

初始有$dp[0][0][0]=1$, 其余赋最大值。

对于转移：

• 若当前字符为1，那么有
  $dp[i][j][0]=\min (dp[i-1][j][0]),$
  $dp[i][j][1]=\min (dp[i-1][j][1])$.
  • 第一个式子不带$dp[i-1][j][1]$是因为$dp[i-1][j][1]$说明第$j$段的末尾是$i-1$, 加上当前字符一定是1，那么这一段的末尾肯定是$i$了，和第三维的0矛盾。
  • 第二个式子不带$dp[i-1][j][0]$是因为对于$dp[i-1][j][0]$, 我们不知道第$j$段的末尾具体在哪，若要让第$j$段的末尾为$i$, 我们不知道是否要进行0变1操作，所以不能转移。
• 若当前字符为0，那么有
  $dp[i][j][0]=\min (dp[i-1][j][0],dp[i-1][j][1]),$
  $dp[i][j][1]=\min (dp[i-1][j][1]+1)$.
  • 第一个式子我们只需要第$j$段的末尾不在$i$, 那末尾可以在$i-1$, 也可以不在$i-1$，所以两个都可以转移。
  • 第二个式子要求第$j$段末尾在$i$, 那么可以在第$j$段末尾在$i-1$的基础上加一次操作，让第$i$个字符变成1.同理对于$dp[i-1][j][0]$, 我们不知道第$j$段具体末尾在哪，所以无法转移。
• 若$i\ge tar$且$j\ge 1$, 有
  $dp[i][j][1]=\min (dp[i-tar][j-1][0]+val[i-tar+1,i])$
  • 即新增一段长度大于等于$tar$的连续1串，操作次数为前$i-tar$个字符、有$j-1$段连续1串，且末尾不在$i-tar$所需的修改次数加上$i-tar+1$到$i$的字符0的个数。

最后只需要判断一下$\min (dp[n][k][0],dp[n][k][1])$是否小于等于$m$即可。

代码

#include <bits/stdc++.h>
using namespace std;

int dp[200003][8][2];

int main() {
	cin.tie(0) -> sync_with_stdio(0);
	int n, m, k;
	string s;
	cin >> n >> m >> k >> s;
	vector<int> cnt_0(n + 1);
	s.insert(0, " ");
	for (int i = 1; i <= n; ++i)
		cnt_0[i] = cnt_0[i - 1] + (s[i] == '0');
	int l = 0, r = 2e5;
	auto check = [&](int tar) {
		for (int i = 0; i <= n; ++i)
			for (int j = 0; j <= k; ++j) dp[i][j][0] = dp[i][j][1] = 0x3f3f3f3f;
		dp[0][0][0] = 0;
		for (int i = 1; i <= n; ++i) {
			for (int j = 0; j <= k; ++j) {
				if (s[i] == '1') {
					dp[i][j][0] = min(dp[i][j][0], dp[i - 1][j][0]);
					dp[i][j][1] = min(dp[i][j][1], dp[i - 1][j][1]);
				}
				else {
					dp[i][j][0] = min({dp[i][j][0], dp[i - 1][j][0], dp[i - 1][j][1]});
					dp[i][j][1] = min(dp[i][j][1], dp[i - 1][j][1] + 1);
				}
				if (j >= 1 && i >= tar && dp[i - tar][j - 1][0] != 0x3f3f3f3f)
						dp[i][j][1] = min(dp[i][j][1], dp[i - tar][j - 1][0] + cnt_0[i] - cnt_0[i - tar]);
			}
		}
		// for (int i = 1; i <= n; ++i) {
  //           for (int j = 0; j <= k; ++j) {
  //               cerr << i << ' ' << j << ' ' << 0 << ' ' << dp[i][j][0] << '\n';
  //               cerr << i << ' ' << j << ' ' << 1 << ' ' << dp[i][j][1] << '\n';
  //           }
  //       }
		return min(dp[n][k][0], dp[n][k][1]) <= m;
	};
	// check(3);
	while (l < r) {
		int mid = (l + r + 1) >> 1;
		// cerr << l << ' ' << r << ' ' << mid << '\n';
		if (check(mid)) l = mid;
		else r = mid - 1;
	}
	if (!l) cout << "-1\n";
	else cout << l << '\n';
	return 0;
}