QUESTION
easy
题目描述
给定一个偶数长度的数组,其中不同的数字代表着不同种类的糖果,每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。
示例 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]
内。
SOLUTION
仔细审题,你只需要找到种类的数目就可以了,如果均分得到的糖果比糖果种类多,那么结果取种类数目;否则取均分可以得到的糖果数量。
方法一
使用集合中不重复元素的特性
class Solution {
public:
int distributeCandies(vector<int>& candies) {
unordered_set<int> s;
for(int e : candies) s.insert(e);
return min(s.size(), candies.size() / 2);
}
};
还可以写的再简洁点:
class Solution {
public:
int distributeCandies(vector<int>& candies) {
unordered_set<int> s(candies.begin(), candies.end());
return min(s.size(), candies.size() / 2);
}
};
方法二
目前在通过的代码里面找到的最快的办法。其中 candy + 100000
是为了满足题意——数组元素的大小范围为 [-100,000, 100,000]
。
class Solution {
public:
int distributeCandies(vector<int>& candies) {
bitset<200001> bs;
for (auto candy : candies) {
bs.set(candy + 100000);
}
return min(bs.count(), candies.size() / 2);
}
};