leetcode904. 水果成篮

904. 水果成篮

#滑动窗口最大，且滑动窗口内只有两种数字
#核心：记录滑动窗口内的种类和出现的次数，这里我们用哈希表
#能用滑动窗口的题，都是要求在一段连续的子数组中
class Solution(object):
    def totalFruit(self, fruits):
        cnt = Counter()

        left = ans = 0
        for right, x in enumerate(fruits):#这里枚举是关键
            cnt[x] += 1
            while len(cnt) > 2:#种类超过2时就应该判断
                cnt[fruits[left]] -= 1#这里，先对left这个种类的个数-1
                if cnt[fruits[left]] == 0:#然后再判断这个种类滑动窗口内数量是不是为0
                    cnt.pop(fruits[left])#如果为0了就移除这个种类
                left += 1#然后顺势向前移动滑动窗口
            ans = max(ans, right - left + 1)
        
        return ans

Counter() 是 collections 库中的一个函数，可以用来统计一个 python 列表、字符串、元组等可迭代对象中每个元素出现的次数，并返回一个字典。使用 Counter 统计完词频后可以使用most_common方法来查找出现频率最高的 n 个数字及其出现次数。还有element() update() subtract()方法