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