时间:1s 空间:256M
题目描述:
小信有一个长为 n 的只含 0 和 1字符串 s。 他可以进行最多 k 次如下操作:
-
选择字符串 s 的一个子串,将其中的字符反转(0 变成 1,1 变成 0)。
求在操作过程中(操作数可为0)出现过的最长的连续的 1 的长度。
输入格式:
第一行,2 个正整数 n,k; 第二行,字符串 s。
输出格式:
输出不超过 k 次操作后,最长的连续的 1 的长度。
样例1输入:
5 1 00010
样例1输出:
4
样例2输入:
5 2 00010
样例2输出:
5
约定:
对于100%的数据,1≤n,k≤105。 字符串 s 只由 0 和 1 组成,长度为 n。
显而易见,每一个字串都只能进行一次翻转
下附代码
#include<bits/stdc++.h>
using namespace std;
vector<pair<int, int>> v;
int s[100005];
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n, k;
string ss;
cin >> n >> k >> ss;
for (int i = 1; i <= n;++i){
s[i] = ss[i - 1] - '0';
}
s[n + 1] = 2;
int c = s[1], sum = 1;
for (int i = 2; i <= n + 1;i++){
if(c!=s[i]){
v.push_back({c, sum});
sum = 1;
c = s[i];
}else{
sum++;
}
}
for (int i = 1; i < v.size();i++){
v[i].second += v[i - 1].second;
}
n = v.size();
sum = 0;
int ans = 0, l = 0;
for (int r = 0; r < n;++r){
if(v[r].first==0){
sum++;
}
while(sum>k){
if(v[l].first==0){
sum--;
}
l++;
}
if(l-1<0){
ans = max(ans, v[r].second);
}
else{
ans = max(ans, v[r].second - v[l - 1].second);
}
}
cout << ans;
}