题目来源于 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;
    }
}




往期热文

拜托,面试官别问我「布隆」了

有点难度,几道和「滑动窗口」有关的算法面试题

荐阅读

几道和「二叉树」有关的算法面试题
几道和「堆栈、队列」有关的面试算法题

链表算法面试问题?看我就够了!

一道看完答案你会觉得很沙雕的「动态规划算法题」

如何有效的写算法题

来和程序员小吴一起学算法~

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


我就知道你在看!