题目描述
给定一个偶数长度的数组,其中不同的数字代表着不同种类的糖果,每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。
示例 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]内。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/distribute-candies
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
可能一下子读不懂题目,其实可以换个角度想想,考虑如果弄清楚了所给的糖果的种类数目的话,这道题又会怎样去解?
其实题目答案就是 min(糖果种类数, 糖果总数/2),因为你可以尽可能地给妹妹分得种类最多的糖果,然后余下的再给弟弟,而妹妹最多也不会分到超过总数一半的糖果,因此有上限。
如何确定种类数目?Set便是极好的选择。
代码详解
class Solution {
public int distributeCandies(int[] candyType) {
HashSet<Integer> hs = new HashSet<>();
int count = 0; // 记录遍历到当前元素时,不同元素的个数
for(int i = 0; i < candyType.length; ++i) {
// 利用Set的不重复添加特性,一旦重复添加则会添加失败,不进入第一个if
if(hs.add(candyType[i])) {
++count;
}
// 此if可以提前结束判断,==也可写做>=
if(count == candyType.length / 2) {
return count;
}
}
return count;
}
}
注意点
- 可以返回Math.min(count, candyType.length / 2),也可以利用上述判断的方式提前结束循环。