ABC229G
首先显然能想到二分,随后想想怎么判定。
这里我卡住了(
看了题解发现是一个常用的技巧。
先把所有 Y 的位置存下来,记为数组 A,记 Bi=Ai−i,那么发现交换就相当于把 B 中某个值加一或减一,接下来就是 check
能否在 K 次内形成一段连续的相同的值,由于 B 是单调不降的,显然取中位数最优,预处理 B 数组前缀和就能轻松求出所需的操作次数。
时间复杂度 O(nlogn)。
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;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话