LeetCode 575. 分糖果(Python)

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

示例 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]内。

 

思路:由于规定输入列表长度为偶数,所以本题较易。

首先要做的就是统计糖果的种类,即去重。去重之后有两种情况:

①糖果种类过多,没办法每一样都获得。则 结果=糖果总数/2 ,因为是偶数,所以弟弟与妹妹糖果数一样,返回总数/2实际也就是按照不同种类拿,每一种一个。

②糖果种类数过少,必定有重复的糖果类别。则 结果=糖果种类数,即确保妹妹有最大种类数。

class Solution(object):
    def distributeCandies(self, candies):
        """
        :type candies: List[int]
        :rtype: int
        """
        d = dict()
        for i in range(len(candies)):
            if not d.__contains__(candies[i]):
                d[candies[i]] = 1
            
        return min(len(d),len(candies)//2)  # 糖果种类过少,糖果种类过多
        

 

你可能感兴趣的:(Python,打卡,Python刷题之路)