LeetCode 575 分糖果

LeetCode 575 分糖果

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/distribute-candies/

博主Github：https://github.com/GDUT-Rp/LeetCode

题目：

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 种糖。

提示：

• $n$ == candyType.length
• $2$ <= $n$ <= $10^4$
• $n$ 是一个偶数
• $-10^5$ <= candyType[i] <= $10^5$

解题思路：

方法一：贪心

直观想法

C++

class Solution {
public:
    int distributeCandies(vector<int> &candyType) {
        return min(unordered_set<int>(candyType.begin(), candyType.end()).size(), candyType.size() / 2);
    }
};

Golang

func distributeCandies(candyType []int) int {
    candyDuplicates := map[int]struct{}{}
    for _, i := range candyType {
        candyDuplicates[i] = struct{}{}
    }
    return min(len(candyDuplicates), len(candyType) / 2)
}


func min(a, b int) int {
    if a > b {
        return b
    } else {
        return a
    }
}

Python

class Solution:
    def distributeCandies(self, candyType: List[int]) -> int:
        candy_duplicate = set(candyType)
        if len(candy_duplicate) >= len(candyType) / 2:
            return len(candyType) // 2
        else:
            return len(candy_duplicate)

更加精简的写法：

class Solution:
    def distributeCandies(self, candyType: List[int]) -> int:
        return min(len(set(candyType)), len(candyType) // 2)

复杂度分析

时间复杂度：$O(n)$
空间复杂度：$O(n)$