哈希表 分糖果

2020-06-16 11:36 更新

题目

难度:简单

给定一个偶数长度的数组,其中不同的数字代表着不同种类的糖果,每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。

示例 1:

输入: candies = [1,1,2,2,3,3] 输出: 3 解析: 一共有三种种类的糖果,每一种都有两个。 最优分配方案:妹妹获得[1,2,3],弟弟也获得[1,2,3]。这样使妹妹获得糖果的种类数最多。

示例 2 :

输入: candies = [1,1,2,3] 输出: 2 解析: 妹妹获得糖果[2,3],弟弟获得糖果[1,1],妹妹有两种不同的糖果,弟弟只有一种。这样使得妹妹可以获得的糖果种类数最多。

注意:

数组的长度为[2, 10,000],并且确定为偶数。 数组中数字的大小在范围[-100,000, 100,000]内。

解法一、利用HashSet的特点

不可重复插入,于是依次插入,寻找共有最大几个不同元素,与length/2,求最小

class Solution {
    public int distributeCandies(int[] candies) {
        HashSet <Integer> set = new HashSet < > ();
        for (int candy : candies){
            set.add(candy);
        }
        return Math.min(set.size(),candies.length/2);
    }
}

时间复杂度:O(n)。整个candiescandiescandies 数组只遍历一次。这里,n表示 candies 数组的大小。 空间复杂度:O(n),在最坏的情况下,set 的大小为 n。

同理在Python中

class Solution:
    def distributeCandies(self, candies: List[int]) -> int:
        return len(set(candies)) if len(set(candies))<=len(candies)//2 else len(candies)//2
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号