LeetCode刷题笔记_575. 分糖果

题目描述

Alice 有 n 枚糖，其中第 i 枚糖的类型为 candyType[i] 。Alice 注意到她的体重正在增长，所以前去拜访了一位医生。

医生建议 Alice 要少摄入糖分，只吃掉她所有糖的 n / 2 即可（n 是一个偶数）。Alice 非常喜欢这些糖，她想要在遵循医生建议的情况下，尽可能吃到最多不同种类的糖。

给你一个长度为 n 的整数数组 candyType ，返回： Alice 在仅吃掉 n / 2 枚糖的情况下，可以吃到糖的最多种类数。

输入：candyType = [1,1,2,3]
输出：2
解释：Alice 只能吃 4 / 2 = 2 枚糖，不管她选择吃的种类是 [1,2]、[1,3] 还是 [2,3]，她只能吃到两种不同类的糖。

思路

吃到糖的最多种类数是 n / 2 和 种类总数的 最小值。
很直接的想到哈希表获得 种类总数，需要一次遍历。

class Solution {
    public int distributeCandies(int[] candyType) {
        Set candies = new HashSet();
        for (int i = 0; i < candyType.length; i++) {
            candies.add(candyType[i]);
        }

        int n = candyType.length / 2;
        return Math.min(n, candies.size());
    }
}

时间和空间复杂度都是O(N)

题解

官方管着叫贪心。和我的差不多。

class Solution {
    public int distributeCandies(int[] candyType) {
        Set<Integer> set = new HashSet<Integer>();
        for (int candy : candyType) {
            set.add(candy);
        }
        return Math.min(set.size(), candyType.length / 2);
    }
}

参考资料

https://leetcode-cn.com/problems/distribute-candies/