三分钟看完「分糖果」算法问题

题目来源于 LeetCode 上第 575 号问题：分糖果。题目难度为 Easy，目前通过率为 60.2% 。

题目描述

给定一个偶数长度的数组，其中不同的数字代表着不同种类的糖果，每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。

示例 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]，妹妹有两种不同的糖果，弟弟只有一种。这样使得妹妹可以获得的糖果种类数最多。

注意:

1. 数组的长度为[2, 10,000]，并且确定为偶数。

2. 数组中数字的大小在范围[-100,000, 100,000]内。

题目解析

总共有 n 个糖，平均分给两个人，每人得到  n／2 块糖，那么能拿到的最大的糖的种类数也就是 n／2 种，不可能更多，只可能更少。

所以只需要统计出糖的种类数，如果糖的种类数小于 n／2，说明拿不到 n／2种糖，最多能拿到的种类数就是当前糖的总种类数。最后，看（数量的一半）和（所有的种类）哪个先达到，取两者中较小的值即可。

举个例子：

• 极端情况1：所有糖都不重样，这种情况妹妹也只能拿到一半的糖果。

• 极端情况2：只有一种糖，这种情况妹妹只能得到一种糖果。

• 平均情况：每个糖都有两个，这种情况妹妹可以拿到所有种类，数量与 极端情况1 一样。

求糖果的种类数使用 hash 即可。

图片描述

代码实现

class Solution {
    public int distributeCandies(int[] candies) {
        Set<Integer> nums = new HashSet<>();
        for (int i = 0; i < candies.length; i++) {
            nums.add(candies[i]);
        }
        int numNums = nums.size();
        int numTarget = candies.length / 2;
        return numNums >= numTarget ? numTarget : numNums;
    }
}