Leetcode 575. 分糖果

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]，妹妹有两种不同的糖果，弟弟只有一种。这样使得妹妹可以获得的糖果种类数最多。
注意
1.数组的长度为[2, 10,000]，并且确定为偶数。
2.数组中数字的大小在范围[-100,000, 100,000]内。

题目解答

从题意要求来看，就是要保证妹妹获得的一半糖果中的糖果种类数目最大。

解题方法

示例中，可能会出现两种情况。
1. 糖果种类数目大于糖果数目的一半，那这个时候妹妹所分到的一半糖果中，每一个都是不同的种类。
2. 糖果种类数目小于等于糖果数目的一般，那这个时候说明妹妹所分到的一半糖果中，会有重复的糖果种类出现，但是却可以包含全部的糖果种类。

代码

class Solution:
    def distributeCandies(self, candies):
        """
        :type candies: List[int]
        :rtype: int
        """
        size = len(candies)
        can_set = set(candies)
        if size / 2 >= len(can_set):
            return len(can_set)
        elif size / 2 < len(can_set):
            return int(size/2)