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 种类数

 

import collections


class Solution(object):
    def distributeCandies(self, candyType):
        """
        :type candyType: List[int]
        :rtype: int
        """
        diff = len(collections.Counter(candyType))
        count = len(candyType)
        if diff >= len(candyType) / 2:
            return count / 2
        else:
            return diff

代码优化,字典部分稍微有点多余,可以改为集合,if也可以缩短为一行

        diff = len(set(candyType))
        lens = len(candyType) / 2
        return lens if diff >= lens else diff

还可以再简化一点

return min(len(set(candyType)), len(candyType) // 2)