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