分糖果-分到最多种类糖果

207 阅读2分钟

分糖果-分到最多种类糖果

题目描述

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

实现


#include <set>

#include <vector>


// cans 长度是 2N 偶数
int m90009(vector<int> cans){
    set<int> set; // 借助 set 容器,存储不重复的元素
    for( int v : cans){
        if(set.size()==cans.size()/2){  // 组多能分到 N/2 一半
            return set.size();
        }
        set.insert(v);
    }
    return set.size();
}

运行结果

main 函数调度运行

#include <iostream>
using namespace std;

#include <vector>


int m90009(vector<int> cans);

int main() {
    std::cout << "尘远同学编程-少年班欢迎您! 报名热线 : 025 - 8662 0610" <<std::endl;

    //vector<int> cans={1,2,3,4,5,6,7,8,9,10};
    // vector<int> cans={1,1,1,1,1,1};
    vector<int> cans={1,1,2,2,3,3,1,2,4};
    cout<<"cans == [";
    for(int i:cans){
        printf(" %d ,",i);
    }
    int kinds =m90009(cans);

    printf("]. kinds ==%2d\n",kinds);

    return 0;


}

image.png