Leetcode 904. Fruit Into Baskets

转载于 2019-03-14 21:47:00 发布 · 94 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zywscq/p/10533747.html

本文详细介绍了滑动窗口算法在水果收集问题中的应用，通过Python实现，展示了如何使用Counter类来跟踪不同类型的水果数量，同时保持窗口内的水果类型不超过三种，以最大化收集的水果数量。

sliding window(滑动窗口)算法

class Solution(object):
    def totalFruit(self, tree):
        """
        :type tree: List[int]
        :rtype: int
        """
        count=collections.Counter()
        left=ans=0
        for i,val in enumerate(tree):
            count[val]+=1
            while (len(count)>=3):
                count[tree[left]]-=1
                if count[tree[left]]==0:
                    del count[tree[left]]
                left+=1
            ans=max(ans,i-left+1)
        return ans