LeetCode 575. 分糖果 Distribute Candies

【问题描述】
给定一个偶数长度的数组,其中不同的数字代表着不同种类的糖果,每一个数字代表一个糖果。你需要把这些糖果平均分给一个弟弟和一个妹妹。返回妹妹可以获得的最大糖果的种类数。
【示例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],妹妹有两种不同的糖果,弟弟只有一种。这样使得妹妹可以获得的糖果种类数最多。

【提示】

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

思路:
1、题目意思是返回最大的种类数,但是不能大于糖果总数的一半
2、种类大于总数的一半,返回总数的一半,else 返回种类数
3、时间复杂度O(n)
4、空间复杂度O(n)

代码实现:
直接用系统的Set,

func distributeCandies(_ candies: [Int]) -> Int {
    let set = Set.init(candies)
    if set.count > candies.count/2 {
        return candies.count/2
    }
    return set.count
}

C语言是如此的快:

#define min(a, b) ((a) < (b) ? (a) : (b))
int distributeCandies(int *candies, int candiesSize) {
    char map[25001] = {0};
    int i, tmp, count = 0;
    for (i = 0; i < candiesSize; i++) {
        tmp = candies[i] + 100000;
        if (!((map[tmp / 8] >> (tmp % 8)) & 1)) {
            map[tmp / 8] |= 1 << (tmp % 8); 
            count = count + 1;
        }   
    }   
    return min(count, candiesSize / 2); 
}

你可能感兴趣的:(LeetCode 575. 分糖果 Distribute Candies)