双指针 方案数

n=7 k=1 只有a,b
abababa
最多修改1次 得到最长连续相同子串的方案数
maxx=3
方案数为5
第一遍双指针求允许修改k个的最长连续子串长度maxx
第二遍判断符合要求的区间有多少个

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
const ll mod=1e9+7;
const int INF=0x3f3f3f3f;
#define IO ios::sync_with_stdio(false);cin.tie(0);
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pb push_back
ll T,n,k,ans,minn=INF;
ll a[maxn];
string s;
ll num;
ll check(char c){
	ll l=0,r=0,num=0,maxx=0,cnt=0;
	while(r<n){
		if(s[r]==c)
			maxx=max(maxx,r-l+1),r++;
		else{
			cnt++;
			while(cnt>k){
				if(s[l]!=c)
					cnt--;
				l++;
			}
			maxx=max(maxx,r-l+1);
			r++;
		}
	}
	return maxx;
}
ll check2(char c){
	ll l=0,r=0,maxx=0,cnt=0,num=0;
	while(r<n){
		if(s[r]==c){
			maxx=max(maxx,r-l+1);
			if(r-l+1==ans)
				num=(num+1)%mod;
			r++;
		}
		else{
			cnt++;
			while(cnt>k){
				if(s[l]!=c)
					cnt--;
				l++;
			}
			maxx=max(maxx,r-l+1);
			if(r-l+1==ans)
				num=(num+1)%mod;
			r++;
		}
	}
	return num%mod;
}
int main(){
	cin>>n>>k;
	cin>>s;
	ans=max(check('a'),check('b'));
	ll ans1=check2('a'),ans2=check2('b');
	cout<<(ans1+ans2)%mod<<endl;
	return 0;
}

牛客变形题目

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+5;
typedef long long ll;
int n,k,ans;
string s;
int check(char c){
	int cnt=0,maxx=0,len=s.size();
	int j=0;
	for(int i=0;i<len;){
		if(j==len)
			break;
		if(s[j]!=c){
		
			cnt++;
			while(cnt>k){
				if(s[i]!=c){
					cnt--;
				}
				i++;
			}
		}
		
		maxx=max(maxx,j-i+1);
		j++;
	}
	return maxx;
}
int main(){
	cin>>n>>k>>s;
	ans=max(check('0'),check('1'));
	cout<<ans<<endl;
	return 0;
}