滑动窗口#

904. 水果成篮

中译中：求只包含两种元素的最长连续子数组

你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果 种类 。

你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：

• 你只有 两个 篮子，并且每个篮子只能装 单一类型 的水果。每个篮子能够装的水果总量没有限制。
• 你可以选择任意一棵树开始采摘，你必须从 每棵 树（包括开始采摘的树）上 恰好摘一个水果 。采摘的水果应当符合篮子中的水果类型。每采摘一次，你将会向右移动到下一棵树，并继续采摘。
• 一旦你走到某棵树前，但水果不符合篮子的水果类型，那么就必须停止采摘。

给你一个整数数组 fruits ，返回你可以收集的水果的 最大 数目。

---

示例 1：

1
输入：fruits = [1,2,1]
2
输出：3
3
解释：可以采摘全部 3 棵树。

示例 2：

1
输入：fruits = [0,1,2,2]
2
输出：3
3
解释：可以采摘 [1,2,2] 这三棵树。
4
如果从第一棵树开始采摘，则只能采摘 [0,1] 这两棵树。

示例 3：

1
输入：fruits = [1,2,3,2,2]
2
输出：4
3
解释：可以采摘 [2,3,2,2] 这四棵树。
4
如果从第一棵树开始采摘，则只能采摘 [1,2] 这两棵树。

---

1.自己的解法(滑动的窗口)

手撕，不过摸爬滚打了好久

1
func totalFruit(fruits []int) int {
2
    left, right := 0, 0
3
  window := make(map[int]int)
4
    count := 0
5
    res := 0
6

7
    if len(fruits) < 3 {
8
        return len(fruits)
9
    }
10

11
    for right < len(fruits) {
12
        window[fruits[right]]++
13
        if window[fruits[right]] == 1 {
14
            count++
15
        }
16

17
        if count < 3 {
18
            res = max(res,right-left+1)
19
        }else{
20
            window[fruits[left]] -= 1
21
            if window[fruits[left]] == 0 {
22
                count--
23
            }
24
            left++
25
        }
26

27
        right++
28
    }
29
    return res
30
}