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],妹妹有两种不同的糖果,弟弟只有一种。这样使得妹妹可以获得的糖果种类数最多。
注意:

数组的长度为[2, 10,000],并且确定为偶数。
数组中数字的大小在范围[-100,000, 100,000]内。

解题思路:
用set()函数判断一共有多少种糖果,如果糖果种类数比每个人分的糖果数多,那么可以给妹妹的糖果全是不同种类的,即为总糖果数的一半;如果糖果种类数小于每个人的糖果数,则将所有不同种类的糖果全部分给妹妹,然后用重复的种类凑够妹妹应得的糖果数,即妹妹获得的种类为全部种类。

class Solution:
    def distributeCandies(self, candies: List[int]) -> int:
        candy=set(candies)
        if len(candy)>=len(candies)/2:
            return int(len(candies)/2)
        else:
            return len(candy)

p.s.也可以用一行来实现

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

猜你喜欢

转载自blog.csdn.net/weixin_44740082/article/details/88858554