题目描述
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 种糖。
题解
- 统计糖果种类:使用一个 unordered_set(无序集合)来存储 candyType 数组中的所有不同糖果种类。由于集合中的元素是唯一的,这可以帮助我们快速统计出不同糖果的种类数。
- 计算糖果总数和种类数:获取 candyType 数组的总长度 totalCandies,以及通过集合 candySet 的大小得到不同糖果的种类数 maxTypes。
- 判断种类数与总数的关系:如果不同糖果的种类数 maxTypes 大于糖果总数的一半 totalCandies / 2,则意味着 Alice 无法吃到所有种类的糖果,因此将 maxTypes 设置为 totalCandies / 2。
- 返回结果:返回 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 可以吃到的糖果种类数,从而高效地解决了问题。