首页 分糖果
文章
取消

分糖果

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

示例 1:

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

示例 2 :

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

注意:

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

分析

总共有 n 个糖,平均分给两个人,每人得到 n/2 块糖,那么能拿到的最大的糖的种类数也就是 n/2 种,不可能更多,只可能更少。

所以只需要统计出糖的种类数,如果糖的种类数小于 n/2,说明拿不到 n/2种糖,最多能拿到的种类数就是当前糖的总种类数。最后,看(数量的一半)和(所有的种类)哪个先达到,取两者中较小的值即可。

举个例子:

极端情况1:所有糖都不重样,这种情况妹妹也只能拿到一半的糖果。

极端情况2:只有一种糖,这种情况妹妹只能得到一种糖果。

平均情况:每个糖都有两个,这种情况妹妹可以拿到所有种类,数量与 极端情况1 一样。

求糖果的种类数使用 hash 即可。

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
func distributeCandies(_ candies: [Int]) -> Int {
        
    var set = Set<Int>()
    for i in candies {

        if !set.contains(i) {

            set.insert(i)
        }
    }
    let nums = set.count
    let target = candies.count / 2  // 最多只能分这么多
    return nums >= target ? target : nums
}

源码地址

本文由作者按照 CC BY 4.0 进行授权

Moya框架浅析

Swift 技术栈