bilibili笔试:给定一个由若干个0和1组成的数组A,我们最多可以将K个值从0变成1,返回仅包含1的最长连续子数组的长度。

比如,输入:

1,1,1,0,0,0,1,1,1,1,0

输出:6。

只要将第三个0和第四个0变成1即可。

思路:回溯,尝试将每个0变成1,一旦将k个0变成1,或者到达字符串的结尾,就开始统计连续的1的子数组的最大长度。

public static int max=0;
	public static void process(int[] arr,int index,int k){
		if(k==0||index==arr.length){//如果已经将k个0变成1了,或者到达了字符串末尾了
			int tmp=0;//开始统计计数连续1的个数
			for(int i=0;i<arr.length;i++){
				if(arr[i]==1){
					tmp++;
					max=Math.max(max, tmp);
				}
				else{//一旦遇到0,之前的计数清零
					tmp=0;
				}
			}
			return ;//结束
		}
		//回溯的思想
		for(int j=index;j<arr.length;j++){
			if(arr[j]==0){
				arr[j]=1;
				process(arr,index+1,k-1);
				arr[j]=0;
			}
		}
	}
	
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		int[] arr={1,1,1,0,0,0,1,1,1,1,0};
		process(arr,0,2);
		System.out.println(max);
		
	}

猜你喜欢

转载自blog.csdn.net/qq_35590091/article/details/108422041