滑动窗口-5--LC904.水果成篮

class Solution(object):
    def totalFruit(self, fruits):
        """
        :type fruits: List[int]
        :rtype: int
        """
        star = 0
        max_len = float('-inf')
        cur_dict = {}
        for end in range(len(fruits)):
            if fruits[end] not in cur_dict:
                cur_dict[fruits[end]] = 1
            else:
                cur_dict[fruits[end]] += 1
            while len(cur_dict) > 2:
                cur_dict[fruits[star]] -= 1
                if cur_dict[fruits[star]] == 0:
                    del cur_dict[fruits[star]]
                star += 1
            max_len = max(max_len, end-star+1)
        return max_len if max_len != float('-inf') else 0