ABC229G

首先显然能想到二分,随后想想怎么判定。

这里我卡住了(

看了题解发现是一个常用的技巧。

先把所有 \(Y\) 的位置存下来,记为数组 \(A\),记 \(B_i=A_i-i\),那么发现交换就相当于把 \(B\) 中某个值加一或减一,接下来就是 check 能否在 \(K\) 次内形成一段连续的相同的值,由于 \(B\) 是单调不降的,显然取中位数最优,预处理 \(B\) 数组前缀和就能轻松求出所需的操作次数。

时间复杂度 \(\mathcal O(n\log n)\)

Code:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 200005;
int n; ll k;
char s[N];
int a[N], b[N], cnt;
ll sum[N];

bool check(int x) {
	for (int l = 1; l <= cnt; ++l) {
		int r = l + x - 1;
		if (r > cnt) break;
		int mid = l + r >> 1; 
		ll tmp = 1ll * (mid - l + 1) * b[mid] - (sum[mid] - sum[l - 1]);
		tmp += (sum[r] - sum[mid - 1]) - 1ll * (r - mid + 1) * b[mid];
		if (tmp <= k) return true;
	}
	return false;
}

int main() {
	scanf("%s", s + 1); n = strlen(s + 1);
	scanf("%lld", &k);
	for (int i = 1; i <= n; ++i) if (s[i] == 'Y') a[++cnt] = i;
	for (int i = 1; i <= cnt; ++i) b[i] = a[i] - i, sum[i] = sum[i - 1] + b[i];
	int l = 0, r = n;
	while (l < r) {
		int mid = l + r + 1 >> 1;
		if (check(mid)) l = mid;
		else r = mid - 1;
	}
	printf("%d", l);
	return 0;
}
posted @ 2022-11-08 20:10  Kobe303  阅读(30)  评论(0编辑  收藏  举报