题目描述
给你一个01字符串,定义答案=该串中最长的连续1的长度,现在你有至多K次机会,每次机会可以将串中的某个0改成1,现在问最大的可能答案。
输入描述
输入第一行两个整数N,K,表示字符串长度和机会次数
第二行输入N个整数,表示该字符串的元素
( 1 <= N <= 300000, 0 <= K <= N )
解题思路
从某一个位置开始,由于K的限制,能够到达的位置是有限的,因此可以考察每一个位置。为了优化遍历,这里有两种方式,首先,可以预先处理得到一个数组sum,用来记录原数组的和,这样sum[j]-sum[i-1]就是i到j位置需要跳动的0的个数,这样使用两个指针l,r,当l,r之间0的个数小于k的时候,移动r进一步扩充,否则,移动l,这样r还可扩充,如果l移动了r的位置,那么从下一个位置开始重新进行上面的过程。
第二个思路是当尽可能的利用K,从某一个位置i开始,如果这个位置是1,移动指针,如果是0,k–,继续移动指针,当k=0的时候,移动到下一个为0的位置j,那么j-i就是使用k次翻转从i位置能够得到的最大连续。然后移动i指针到下一个为0的位置,遇到0之后移动一次指针,就说明了j有了一次翻转0的机会,j还是遇到1直接跳过,如果不是1而是0,i移动带来的额外那个正好可以弥补这个0,所以这个思路就是第一次找到满足条件的位置i,j之后,后面全是在这个基础上进行寻找。
实现代码
思路一:
#include <iostream>
using namespace std;
int max(int a,int b){
return (a>b?a:b);
}
int main(){
int n,k;
cin>>n>>k;
vector<int> sum(n+1,0);
sum[0]=0;
int i,j,val;
for(i=1;i<n;i++){
cin>>val;
sum[i]=sum[i-1]+val;
}
i=1;j=1;int s=0;
while(i<n&&j<n){
if((j-i+1)-sum[j]-sum[i-1]<=k){
s=max(s,j-i+1);
j++;
}else if(i<j)
i++;
else{
i++;
j++;
}
}
}
思路二:
#include <iostream>
#include <vector>
using namespace std;
vector<int> data;
int max(int a,int b){
return a>b?a:b;
}
int main(){
int n,k;
cin>>n>>k;
data.resize(n,0);
for(int i=0;i<n;i++)
cin>>data[i];
int l=0,r=0,maxv=0;
while(r<n){
if(data[r]==1)
r++;
else if(k>0){
k--;
r++;
}else{
maxv=max(maxv,r-l);
while(l<n&&data[l]==1)
l++;
l++;
r++;
}
}
cout<<maxv<<endl;
return 0;
}