Python Leetcode(575.分糖果)

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