【Leetcode】575. 分糖果

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);
    }
};