最长全1串

题目描述

  给你一个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;
}