20191023：（leetcode习题）最大连续1的个数 III

题目

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

输入：A = [1,1,1,0,0,0,1,1,1,1,0], K = 2
输出：6
解释： 
[1,1,1,0,0,1,1,1,1,1,1]
粗体数字从 0 翻转到 1，最长的子数组长度为 6。
输入：A = [0,0,1,1,0,0,1,1,1,0,1,1,0,0,0,1,1,1,1], K = 3
输出：10
解释：
[0,0,1,1,1,1,1,1,1,1,1,1,0,0,0,1,1,1,1]
粗体数字从 0 翻转到 1，最长的子数组长度为 10。

大致思路

1. 滑动窗口即可，当窗口中的0的个数大于K时，将窗口左移，左移时看移出去的时候是1还是0，是0记得将0的个数减一，是1则不需要，最终将滑动窗口的最大值输出即可。

代码实现

package com.immunize.leetcode.longestOnes;

/**
 * 最大1的个数
 * 
 * @author Mr IMMUNIZE
 *
 */
public class Solutioin {
	public static int longestOnes(int[] A, int K) {
		if (A.length == 0)
			return 0;
		int left = 0;
		int count = 0;
		int max = 0;
		for (int i = 0; i < A.length; i++) {
			// 逢0计数加一
			if (A[i] == 0) {
				count++;
			}
			// 超过K时 若左边界为0,则需要移动左边界,出0时记得将计数-1,若为1，则直接右移即可。
			while (count > K) {
				if (A[left] == 0) {
					left++;
					count--;
				} else {
					left++;
				}
			}
			// 更新max即可
			max = Math.max(max, i - left + 1);
		}
		return max;
	}
}