【C++刷题】力扣-#575-分糖果

题目描述

Alice 有 n 枚糖，其中第 i 枚糖的类型为 candyType[i]。根据医生的建议，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 种糖。

题解

1. 统计糖果种类：使用一个 unordered_set（无序集合）来存储 candyType 数组中的所有不同糖果种类。由于集合中的元素是唯一的，这可以帮助我们快速统计出不同糖果的种类数。
2. 计算糖果总数和种类数：获取 candyType 数组的总长度 totalCandies，以及通过集合 candySet 的大小得到不同糖果的种类数 maxTypes。
3. 判断种类数与总数的关系：如果不同糖果的种类数 maxTypes 大于糖果总数的一半 totalCandies / 2，则意味着 Alice 无法吃到所有种类的糖果，因此将 maxTypes 设置为 totalCandies / 2。
4. 返回结果：返回 Alice 可以吃到的糖果种类数 maxTypes。

代码实现

int distributeCandies(vector<int>& candyType) {
    unordered_set<int> candySet(candyType.begin(), candyType.end());
    int totalCandies = candyType.size();
    int maxTypes = candySet.size();
    if (maxTypes > totalCandies / 2) {
        maxTypes = totalCandies / 2;
    }
    return maxTypes;
}

复杂度分析

● 时间复杂度：O(n)，其中 n 是数组 candyType 的长度。我们需要遍历一次数组来构建集合。
● 空间复杂度：O(k)，其中 k 是糖果种类的数量。我们使用集合来存储所有不同的糖果种类。
这个算法的优势在于它直接使用集合来统计不同糖果的种类数，并通过简单的条件判断来确定 Alice 可以吃到的糖果种类数，从而高效地解决了问题。