力扣题解:分糖果

146 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第17天,点击查看活动详情

题目描述

原题链接 :

575. 分糖果 - 力扣(LeetCode)

Alicen 枚糖,其中第 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

思路分析

  • 统计糖果种类数量,借助set的唯一性
  • 因为糖果总数为偶数,且均分,弟弟和妹妹最后分得的糖果数量相同且为糖果数量的一半
  • 如果糖果种类数量小于等于糖果数量的一半,则妹妹能分到一半数量的糖果,每次尽可能的分给其不同的种类
  • 如果糖果种类数量大于糖果数量的一半,但由于最多只能分到一半数量的糖果,所以最多能分到的糖果种类也最多为一半数量的糖果

AC 代码

class Solution {
public:
    int distributeCandies(vector<int>& candyType) {
        int maxCandyTypeNumber = 0;
        // 统计糖果种类
        unordered_set<int> myset(candyType.begin(), candyType.end());
        // 如果糖果种类数小于等于糖果数量的一半
        if(myset.size() <= candyType.size() / 2){
            maxCandyTypeNumber = myset.size();
        } else {
            maxCandyTypeNumber = candyType.size() / 2;
        }
        return maxCandyTypeNumber;
    }
};