给定一个偶数长度的数组,其中不同的数字代表着不同种类的糖果,每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。
示例 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)