575. 分糖果
阿新 • • 發佈:2021-11-01
給定一個偶數長度的陣列,其中不同的數字代表著不同種類的糖果,每一個數字代表一個糖果。你需要把這些糖果平均分給一個弟弟和一個妹妹。返回妹妹可以獲得的最大糖果的種類數。
示例 1:
輸入: candies = [1,1,2,2,3,3]
輸出: 3
解析: 一共有三種種類的糖果,每一種都有兩個。
最優分配方案:妹妹獲得[1,2,3],弟弟也獲得[1,2,3]。這樣使妹妹獲得糖果的種類數最多。
示例 2 :
輸入: candies = [1,1,2,3]
輸出: 2
解析: 妹妹獲得糖果[2,3],弟弟獲得糖果[1,1],妹妹有兩種不同的糖果,弟弟只有一種。這樣使得妹妹可以獲得的糖果種類數最多。
注意:
陣列的長度為[2, 10,000],並且確定為偶數。
陣列中數字的大小在範圍[-100,000, 100,000]內。
思路:注意到儘可能多分不同種類的糖果給妹妹,那麼糖果的種類數就是一個關鍵點.
先求出糖果的種類數
if 種類數 >= 妹妹要分到的糖果數 [說明糖果種類很多,足夠給妹妹分n個不同的糖果]
return 妹妹要分到的糖果數
else [說明糖果種類不夠分給妹妹n個不同的糖果,那就儘可能多分一些不同種類的給她]
return 種類數
1 import collections 2 3 4 class Solution(object): 5 defdistributeCandies(self, candyType): 6 """ 7 :type candyType: List[int] 8 :rtype: int 9 """ 10 diff = len(collections.Counter(candyType)) 11 count = len(candyType) 12 if diff >= len(candyType) / 2: 13 return count / 2 14 else: 15 return diff
程式碼優化,字典部分稍微有點多餘,可以改為集合,if也可以縮短為一行
1 diff = len(set(candyType)) 2 lens = len(candyType) / 2 3 return lens if diff >= lens else diff
還可以再簡化一點
1 return min(len(set(candyType)), len(candyType) // 2)