CF1693F

首先可以假定整个序列 \(c_0\ge c_1\),否则我们把 \(0\) 变成 \(1\)\(1\) 变成 \(0\),并翻转序列。新序列答案与原序列相同。
结论:仅操作 \(c_0=c_1\) 的区间的最小答案和原题一致。
证明:考虑操作了一个 \(c_0-c_1=d\) 的区间。区间第一个数,若是 \(0\),那我们的操作区间其实不应该包含它。也就是说第一个数是 \(1\),那么我们找到一个 \(1\)\(0\) 数量相等的前缀将其排序,第一个数就变成了 \(0\)。然后我们把它去掉,就花费了 \(1\) 的代价进入了 \(d-1\) 的子问题,直到 \(d=0\) 可以一次性排序。这就相当于把原操作拆成了 \(d+1\)\(c_0=c_1\) 的操作。
得知这个结论以后,就可以考虑贪心了。
贪心策略:每次操作一个最长的 \(c_0=c_1\) 的前缀,然后把其中的 \(0\) 删掉。若此时 \(c_0\le c_1\) 了,就可以从删掉的 \(0\) 里面补回来一些,然后一次排好,否则重复此过程。
正确性可以感性理解,在最后一次操作前,前 \(c_0-c_1\) 个数应当全是 \(0\)。那么第一个 \(1\) 肯定是需要往后,而且肯定是越早越好。
具体实现,我们求出 \(mx_i\) 表示最大使得每个前缀中 \(0\)\(1\)\(i\) 个的位置。就可以在每次删掉若干个 \(0\) 后求出下一段 \(01\) 个数相等的最长前缀了。

Code:

#include <bits/stdc++.h>
using namespace std;
const int N = 200005;
int T;
int n;
int a[N]; char s[N];
int mx[N];

void solve() {
	scanf("%d%s", &n, s);
	int c0 = 0, c1 = 0;
	for (int i = 0; i < n; ++i) a[i] = s[i] - '0', c0 += a[i] == 0;
	c1 = n - c0;
	if (!c1) return printf("%d\n", 0), void();
	for (int i = 0; i < n; ++i) {
		bool flag = 1;
		if (a[i] == 1) {
			for (int j = i + 1; j < n; ++j) flag &= a[j];
			if (flag) return printf("%d\n", 0), void();
			break;
		}
	}
	if (c0 < c1) {
		swap(c0, c1), reverse(a, a + n);
		for (int i = 0; i < n; ++i) a[i] ^= 1;
	}
	memset(mx, 0, sizeof (int) * (n + 1));
	for (int i = 0, cur = 0; i < n; ++i) {
		if (a[i]) --cur; else ++cur;
		if (cur >= 0) mx[cur] = i;
	}
	int delta = c0 - c1, cur = 0;
	while (a[cur] == 0 && cur < n) ++cur;
	int ans = 0;
	while (cur < delta) {
		++ans;
		cur += (mx[cur] - cur + 1) / 2;
	}
	printf("%d\n", ans + 1);
}

int main() {
	scanf("%d", &T);
	while (T--) solve();
	return 0;
}
posted @ 2022-09-28 20:08  Kobe303  阅读(18)  评论(0编辑  收藏  举报