Python Leetcode(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]内。
Solution:(看了解析,思路如下:分发糖果,如果每个种类都给妹妹一个之后还有剩余,那么妹妹获得糖果的种类数也就是这些种类的个数;另一种情况是,还没将每种糖果发给妹妹的时候,妹妹的糖果数量已经到了总糖果数量的一半,那么此时妹妹获得糖果的种类数就是总糖果数量的一半。所以要点是看这两种情况谁先达到,用min()方法。)
class Solution(object):
def distributeCandies(self, candies):
"""
:type candies: List[int]
:rtype: int
"""
#set(candies)表示糖果的种类数,len(candies)/2表示糖果总数的一半,若
#糖果种类小,则直接返回种类个数,因为剩下的全是重复的;若总数的一半小,说明
#种类还有剩余,但妹妹无法再拿了,最多只能由总数的一半的种类个数
return min(len(set(candies)), len(candies)/2)
solution = Solution()
print(solution.distributeCandies([1,1,1,1,2,2,2,3,3,3]))
3