力扣每日一题 水果成篮

Day 22

Date: October 17, 2022 8:46 AM
LinkedIn: https://leetcode.cn/problems/fruit-into-baskets/description/
Title: 水果成篮

直接遍历，此时发现当fruits特别大时会超时。

class Solution:
    def totalFruit(self, fruits: List[int]) -> int:
        
        n = len(fruits)
        mx = 0
        flag = 0
        for i in range(n):# 从i 开始采摘
            busket = [1.1]*2 # 篮子初始化
            busket[0] = fruits[i] 
            cnt = 1 # 初始化篮子中水果计数器
            for j in range(i+1, n): # 从后续的每棵树都恰好摘一个
                if fruits[j] == busket[0]: # 与第一个篮子相同
                    cnt += 1
                else:
                    if busket[1] == 1.1:
                        busket[1] = fruits[j] # 第二个篮子还未装一种
                        cnt += 1
                    elif int(busket[1]) == fruits[j]: # 与第二个篮子相同
                        cnt += 1 
                        
                    else:
                       # 停止采摘
                        break
            mx = max(mx, cnt)
        return mx

哈希表 + 滑动窗口，用双指针代表窗口的左右边界，窗口大小即为收集水果的数目

Counter() 是 collections 库中的一个函数，可以用来统计一个 python 列表、字符串
、元组等可迭代对象中每个元素出现的次数，并返回一个字典。

class Solution:
    def totalFruit(self, fruits: List[int]) -> int:
        cnt = Counter() # # 创建一个新的空counter
        ans = j = 0 # ans为收集水果的最大数目，j为窗口左边界
        for i, x in enumerate(fruits):
            cnt[x] += 1 # 对应水果种类的水果数目计数
            while len(cnt) > 2: # 篮子中水果种类大于两种
                y = fruits[j] # 左边界的水果种类
                cnt[y] -= 1 # 将其除去
                if cnt[y] == 0: 
                    cnt.pop(y)
                j += 1 # 左边界前移
            ans = max(ans, i - j + 1) # 更新最大收集水果数目 i为窗口右边界
        return ans