这是我参与11月更文挑战的第1天,活动详情查看:2021最后一次更文挑战
575. 分糖果
给定一个偶数长度的数组,其中不同的数字代表着不同种类的糖果,每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。
示例 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]内
解题思路
贪心的思路,妹妹尝试每种糖果都拿一遍,妹妹最多能拿糖果的数量就是糖果数组的一半,如果糖果种类大于数组的一半,说明妹妹可以拿到的糖果数量每一种都是不同的,糖果数量就是数组长度的一半,否则的话最多只能把所有糖果种类都拿一遍
排序的思路:我们可以对给定的糖果数组进行排序,并通过比较排序数组的相邻元素来找出唯一的元素。对于找到的每个新元素(与前一个元素不同),我们需要更新 cnt。最后,我们可以将所需结果返回为min(n/2,cnt)。
代码
class Solution {
public:
int distributeCandies(vector<int>& candyType) {
unordered_set<int> set;
for (auto a:candyType) set.insert(a);
return set.size()>candyType.size()/2?candyType.size()/2:set.size();
}
};
class Solution {
public:
int distributeCandies(vector<int>& candyType) {
sort(candyType.begin(),candyType.end());
int cnt=1;
for (int i = 1; i < candyType.size(); ++i) {
if (candyType[i]!=candyType[i-1])
cnt++;
}
return cnt>candyType.size()/2?candyType.size()/2:cnt;
}
};