LeetCode刷题(79)~分糖果【哈希】
程序员文章站
2022-06-07 09:58:20
...
题目描述
给定一个偶数长度的数组,其中不同的数字代表着不同种类的糖果,每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。
示例 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]内。
解答 By 海轰
提交代码(哈希)
int distributeCandies(vector<int>& candies) {
unordered_map<int,int> m;
for(int candie:candies)
{
++m[candie];
}
int count=0;
for(unordered_map<int,int>::iterator it=m.begin();it!=m.end();++it)
{
++count;
if(count==candies.size()/2)
break;
}
return count;
}
运行结果
提交代码(哈希-优化)
int distributeCandies(vector<int>& candies) {
unordered_map<int,int> m;
int count=0;
for(int candie:candies)
{
++m[candie];
if(m[candie]==1)
++count;
if(count==candies.size()/2)
break;
}
return count;
}
运行结果
提交代码(排序)
int distributeCandies(vector<int>& candies) {
sort(candies.begin(),candies.end());
int count=1;
for(int i=1;i<candies.size()&&count<candies.size()/2;++i)
{
if(candies[i]>candies[i-1])
++count;
}
return count;
}
运行结果
提交代码(unordered_set)
int distributeCandies(vector<int>& candies) {
unordered_set<int> s;
for(int candie:candies)
s.insert(candie);
return min(s.size(),candies.size()/2);
}
运行结果
思路:尽可能先让妹妹选完每一个种类的糖果(前提是自己的糖果数量小于等于一半)
题目来源
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/distribute-candies