题意: 给定一个字符串,最多能改变k个字母,求使得字母都相同的子字符串的最大长度(字符串只有a或者b组成)
1. 二分
思路: 这个子字符串既可以由相同的a组成,也可以由完全相同的b组成。枚举这个字符串的左端点,二分这个字符串的右端点,找到最长的长度。
#include <iostream>
#include <algorithm>
using namespace std;
const int M = 1e5 + 5;
char ch[M];
int sum[M];
int n,k;
int check(int a,int b)
{
if(sum[a] - sum[b] <= k) return 1;
return 0;
}
int main()
{
cin >> n >> k;
cin >> ch;
sum[0] = 0;
for(int i = 0;i < n;i ++)//判断字符串中该位置之前有多少字符b
{
if(ch[i] == 'a') sum[i+1] = sum[i];
else sum[i+1] = sum[i] + 1;
}
int ans = 0;///初始化长度为0
for(int i = 0;i < n; i++)///左端点枚举
{
int l = i, r = n;///右端点二分,右端点从左端点到最大值 变化进行二分
while(l <= r)
{
int mid = (l + r)/2;
if(check(mid, i)) l = mid + 1;
else r = mid - 1;
}
ans = max(ans ,l - i - 1);///更新长度的最大值
}
for(int i = 0;i < n;i++)///找当前位置之前的字符a的个数
{
if(ch[i] == 'b') sum[i+1] = sum[i];
else sum[i+1] = sum[i] + 1;
}
for(int i = 0;i < n;i++)
{
int l = i, r = n;
while(l <= r)
{
int mid = (l + r)/2;
if(check(mid ,i)) l = mid + 1;
else r = mid - 1;
}
ans = max(ans, l - i - 1);
}
cout << ans << endl;
return 0;
}
2.尺取
#include <iostream>
#include <algorithm>
using namespace std;
const int M = 1e5 + 5;
char ch[M];
int n,k;
int a[M];
int func()
{
int s = 0,t = 0,ans = 0;
int cnt = 0;
while(s <= t)
{
while(cnt <= k && t < n)
{
if(a[t] == 0) cnt ++;
if(cnt <= k) ans = max(ans, t-s+1);
t++;
}
if(a[s] == 0) cnt --;
s ++;
}
return ans;
}
int main()
{
cin >> n >> k;
cin >> ch;
for(int i = 0;i < n;i++)
a[i] = ch[i] - 'a';
int ans1 = func();
for(int i = 0;i < n;i++)
a[i] = a[i] ? 0 : 1;
int ans2= func();
cout << max(ans1,ans2) << endl;
return 0;
}