575. 分糖果
难度 简单
题目描述: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 种糖。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/distribute-candies
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
提示:
- n == candyType.length
- 2 <= n <= 104
- n 是一个偶数
- -105 <= candyType[i] <= 105
自己想到的本方法
class Solution {
public:
int distributeCandies(vector<int>& candyType) {
map<int, int>candy;
int candyCnt = 0;
for (auto x:candyType) {
if (candy.find(x) == candy.end()) {
candy.insert(pair<int, int>(x,1));
} else {
candy[x]++;
}
}
int cntt = 0;
map<int, int>::iterator it;
for (it = candy.begin(); it != candy.end(); it++) {
if (it->second >= 2) {
candyCnt++;
it->second -= 2;
cntt += it->second;
it->second = 0;
}
}
int size = 0;
for (auto x:candy) {
if (x.second > 0){
size++;
}
}
if (cntt >= size) {
return candyCnt + size;
} else {
return candyCnt + cntt + (size-cntt)/2;
}
}
};
解析:最大极端情况,所有糖都不重样,那么妹妹也只能得到一半。 中间情况,每个糖都有两个,那么妹妹刚好能得到所有种类,数量跟第一种一样。 最小极端情况,就一种糖,那么妹妹就只能得到一种。
思想,不管有多少种,先每种一个给妹妹,如果才给了妹妹几个,所有的种类就都给过了,那后面的都是重复的了。如果给了妹妹一半的数量,还有不重样的,妹妹最多也就只有一半数量的种类。所以说到最后,看(数量的一半)和(所有的种类)哪个先达到,也就是取两者中较小的值。
// 用集合去重 比较
class Solution {
public:
int distributeCandies(vector<int>& candyType) {
set<int> candySet;
for (auto x:candyType) {
candySet.insert(x);
}
return min(candySet.size(), candyType.size()/2);
}
};
class Solution {
public:
int distributeCandies(vector<int>& candyType) {
set<int> candySet;
for (auto x:candyType) {
candySet.insert(x);
if (candySet.size() == candyType.size()/2)
return candyType.size()/2; //提前break
}
return min(candySet.size(), candyType.size()/2);
}
};