【力扣】575. 分糖果

507 阅读2分钟

小知识,大挑战!本文正在参与「程序员必备小知识」创作活动

原题链接

575. 分糖果 - 力扣(LeetCode) (leetcode-cn.com)

题目描述

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

测试用例

示例 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]内。

分析

题目可以这么理解,有一个偶数长的数组,我们把它均分为长度相等的两份,其中一份中拥有的最多个不同的元素个数是多少。即我们需要在原数组中尽可能的提取出不同的元素放入到 len/2 长度的子数组中

我们可以使用 Set 这个数据结构,最后得到的长度,最多就是 len/2,如果种类不够,那就是 set.size

代码

根据上面的分析,一行代码写完

var distributeCandies = function(candyType) {
    return Math.min(candyType.length/2,new Set(candyType).size);
};

image.png

解法 2

上一种解法写起来简单,但仔细思考,就会发现,我们需要吧整个数组转化为 Set 再来做判断,这里可以优化一下空间占用

我们逐个遍历原始数组,吧元素添加到 Set,一旦 set.size == len/2,我们就没必要继续去处理后续的数据了(毕竟子数组的长度限定了 len/2)

var distributeCandies = function(candyType) {
    let set = new Set();
    let half = candyType.length / 2;
    for (let i = 0; i < candyType.length; i++) {
        set.add(candyType[i]);
        if (set.size >= half) return half;
    }
    return set.size;
};

image.png

最差的情况也就是全部遍历一次数组,最好的情况,只需要遍历一半的数组,代码也证明了确实对内存消耗有不少的优化


今天的力扣刷题就分享到这里,感谢大家的阅读~