1. 程式人生 > 其它 >575. 分糖果

575. 分糖果

給定一個偶數長度的陣列,其中不同的數字代表著不同種類的糖果,每一個數字代表一個糖果。你需要把這些糖果平均分給一個弟弟和一個妹妹。返回妹妹可以獲得的最大糖果的種類數。

示例 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     def
distributeCandies(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)