LeetCode 1004. 最大连续1的个数 III

给定一个由若干 0 和 1 组成的数组 A，我们最多可以将 K 个值从 0 变成 1 。

返回仅包含 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。

示例 2：

输入：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. 1 <= A.length <= 20000
2. 0 <= K <= A.length
3. A[i] 为 0 或 1 

 

//1004. 最大连续1的个数 III
func longestOnes(A []int, K int) int {
	max := 0
	for i := 0; i < len(A); i++ {
		count := 0
		k := K
		for _, s := range A[i:] {
			if s == 0 && k > 0 {
				k--
				s = 1
			}
			if s == 0 {
				break
			}
			count++
		}
		if count > max {
			max = count
		}
	}
	return max
}