2020牛客寒假算法基础集训营1 H.nozomi和字符串
题目描述
nozomi看到eli在字符串的“花园”里迷路了,决定也去研究字符串问题。
她想到了这样一个问题:
对于一个 “01”串而言,每次操作可以把 0 字符改为 1 字符,或者把 1 字符改为0 字符。所谓“01”串,即只含字符 0 和字符1 的字符串。
nozomi有最多k次操作的机会。她想在操作之后找出一个尽可能长的连续子串,这个子串上的所有字符都相同。
nozomi想问问聪明的你,这个子串的长度最大值是多少?
注: 次操作机会可以不全部用完。
如果想知道连续子串的说明,可以去问问eli,nozomi不想再讲一遍。
输入描述:
第一行输入两个正整数 和
输入仅有一行,为一个长度为 的、仅由字符 0 和 1 组成的字符串。
输出描述:
一个正整数,为满足条件的子串长度最大值。
示例1
输入
5 1
10101
输出
3
这题就是用两个vector分别记录下每个0和1的位置,假设1变0,变换k次即求第i个1的前缀与第i+k-1个1的后缀的距离的最大值,须将第一个1的前缀置为-1,最后一个的后缀置为n,以应对k=0的情况,AC代码如下:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
int n,k;
string s;
vector<int>q0,q1;
cin>>n>>k>>s;
q0.push_back(-1);
q1.push_back(-1);
for(int i=0;i<n;i++){
if(s[i]=='0') q0.push_back(i);
else q1.push_back(i);
}
q0.push_back(n);
q1.push_back(n);
int ans=0;
if(q0.size()-2<=k) ans=n;
else{
for(int i=1,j=k;j<q0.size()-1;i++,j++)
ans=max(ans,q0[j+1]-q0[i-1]-1);
}
if(q1.size()-2<=k) ans=n;
else{
for(int i=1,j=k;j<q1.size()-1;i++,j++)
ans=max(ans,q1[j+1]-q1[i-1]-1);
}
cout<<ans;
return 0;
}