分糖果
题目描述:
Alice 有 n 枚糖,其中第 i 枚糖的类型为 candyType[i] 。Alice 注意到她的体重正在增长,所以前去拜访了一位医生。
医生建议 Alice 要少摄入糖分,只吃掉她所有糖的 n / 2 即可(n 是一个偶数)。Alice 非常喜欢这些糖,她想要在遵循医生建议的情况下,尽可能吃到最多不同种类的糖。
给你一个长度为 n 的整数数组 candyType,返回:Alice在仅吃掉 n / 2 枚糖的情况下,可以吃到糖的最多种类数。
示例 1:
输入:candyType = [1,1,2,2,3,3]
输出:3
解释:Alice 只能吃 6 / 2 = 3 枚糖,由于只有 3 种糖,她可以每种吃一枚。
示例 2:
输入:candyType = [1,1,2,3]
输出:2
解释:Alice 只能吃 4 / 2 = 2 枚糖,不管她选择吃的种类是 [1,2]、[1,3] 还是 [2,3],她只能吃到两种不同类的糖。
示例 3:
输入:candyType = [6,6,6,6]
输出:1
解释:Alice 只能吃 4 / 2 = 2 枚糖,尽管她能吃 2 枚,但只能吃到 1 种糖。
方法一:贪心
一方面,设糖果数量为 n,由于妹妹只能分到一半的糖果,所以答案不会超过 n/2;另一方面,设这些糖果一共有 m 种,答案也不会超过 m。
若 m≤n/2,则可以每种糖果至少分一颗给妹妹,此时答案为 m;若 m>n/2 ,则妹妹只能分到 n/2种糖果,每种糖果分一颗,此时答案为 n/2 。
综上所述,答案为 min(m,n/2)。
代码详情:
class Solution:
def distributeCandies(self, candyType: List[int]) -> int:
return min(len(set(candyType)),len(candyType)//2)
备注:
min()函数:取最小值;
len()函数:计算数组长度
set()函数:创建一个无序不重复元素集
candyType: List[int]:candyType为整型列表
-> int:返回结果应为整型
原作者:力扣官方题解
链接:https://leetcode.cn/problems/distribute-candies/solutions/1072396/fen-tang-guo-by-leetcode-solution-l4f6/
来源:力扣(LeetCode)