题目描述
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 == candyType.length
2 <= n <= 104
n 是一个偶数
-105 <= candyType[i] <= 105
思路:
贪心
根据题意,妹妹每种糖果至少要拿一颗1.
- 如果糖果种类数<=数组长度的一半,那么妹妹可以拿到所有种类的糖果。ans = 糖果种类数
- 如果糖果种类数>数组长度的一半,那么妹妹能拿到的所有糖果都互不相同。ans = candyType.length/2
二者取小即可。
class Solution {
public int distributeCandies(int[] candyType) {
//种类数大于等于n 答案等于n。
//种类数小于n 答案等于种类数。
Set<Integer> set = new HashSet<>();
for (int c: candyType) set.add(c);
return Math.min(set.size(), candyType.length>>1);
}
}