Description:

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

返回仅包含 1 的最长（连续）子数组的长度。

Examples:

示例 1：

1
2
3
4
5
输入：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：

1
2
3
4
5
输入：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。

Solution:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
func longestOnes(A []int, K int) int {
    size := len(A)
    left, right := 0, 0
    res, sum := 0, 0
    for right < size {
        sum += A[right] ^ 1
        for sum > K {
            sum -= (A[left] ^ 1)
            left += 1
        }

        res = max(res, right-left+1)
        right += 1
    }
    return res
}

func max(a,b int) int{
    if a > b{
        return a
    }
    return b
}