题目:
Alice 有 n
枚糖,其中第 i
枚糖的类型为 candyType[i]
。Alice 注意到她的体重正在增长,所以前去拜访了一位医生。
医生建议 Alice 要少摄入糖分,只吃掉她所有糖的 n / 2
即可(n
是一个偶数)。Alice 非常喜欢这些糖,她想要在遵循医生建议的情况下,尽可能吃到最多不同种类的糖。
给你一个长度为 n
的整数数组 candyType
,返回: Alice 在仅吃掉 n / 2
枚糖的情况下,可以吃到糖的 最多 种类数。
思路:
用一个集合visited记录哪些糖果种类吃过,遍历所有的糖果:当当前糖果种类不在visited中时,ans加一,并将该种类加入到visited中。当ans等于n/2时,或遍历完所有糖果时,结束循环。代码如下:
class Solution:def distributeCandies(self, candyType: List[int]) -> int:n = len(candyType)ans = 0index = 0 # 糖的索引visited = set() # 集合记录已吃过的种类while ans < n/2 and index < n:itype = candyType[index]if itype not in visited:ans += 1visited.add(itype)index += 1return ans
提交通过:
优化:
从例子中可以发现规律:设一共有k种糖,糖的总数为n,若k>n/2,说明吃到的所有糖都是不同种类,答案为n/2;若k<n/2,说明吃到的糖里只有k种,答案为k。也就是说答案为min(k,n/2)。代码如下:
class Solution:def distributeCandies(self, candyType: List[int]) -> int:n = len(candyType)itype = set(candyType) # 利用集合去掉重复种类k = len(itype)return min(k, int(n/2))
提交通过: