【LeetCode】分糖果Java 题解

168 阅读2分钟

这是我参与11月更文挑战的第1天,活动详情查看:2021最后一次更文挑战

题目描述

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 ,返回: Alice 在仅吃掉 n / 2 枚糖的情况下,可以吃到糖的最多种类数。 根据这个重要提示,我们首先采用 hashSet 这种数据结构,统计所有糖果的种类 m,然后根据可以吃的最多糖果个数 n / 2, 与糖果种类比较。如果n / 2 小于等于 m, 则最多吃 n / 2 种类,反之,最多吃 m 种类。 具体实现代码如下:

通过代码

class Solution {
    public int distributeCandies(int[] candyType) {
        int n = candyType.length;
        int ans = 0;
        Set<Integer> set = new HashSet<>();
        for (int candy : candyType) {
            set.add(candy);
        }

        int m = set.size();
        if (n / 2 <= m) {
            ans = n / 2;
        } else {
            ans = m;
        }

        return ans;
        
    }
}

image.png

总结

  • 上述解法的时间复杂度是O(n), 空间复杂度是O(n)
  • 坚持算法每日一题,加油!