904. 水果成篮

最新推荐文章于 2024-10-01 05:04:32 发布

Marshal～

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量111

点赞数

分类专栏：刷题

本文链接：https://blog.csdn.net/M_arshal_/article/details/123621895

版权

Python collections.Counter 循环优化计数问题列表处理

关键词由CSDN通过智能技术生成

刷题专栏收录该内容

16 篇文章 0 订阅

订阅专栏

在这里插入图片描述
主要是用了Counter这个类，以及在循环中的操作

import collections
class Solution:
    def totalFruit(self, fruits: List[int]) -> int:
        count= collections.Counter()
        ans,i=0,0
        for j,x in enumerate(fruits):
            count[x]+=1
            while len(count)>=3:
                count[fruits[i]]-=1
                if count[fruits[i]]==0:
                    del count[fruits[i]]
                i+=1
            ans = max(ans,j-i+1)
        return ans