找出一个数组内,出现次数大于n/2次的元素值。
Naive解法,将值放入hashtable内,并计数。14ms,排名不前,应该是我的hashfunction(Murmur3)对于一个int来说太过复杂了。
struct Mystruct {
int key;
int num;
struct B_HashHandle hh;
};
int majorityElement(int* nums, int numsSize) {
<pre name="code" class="cpp"> if (numsSize == 1) return nums[0]; // 判断只有一个元素的情况
int i, t = 0;
struct Mystruct* mystruct = malloc(sizeof(struct Mystruct) * numsSize);
struct B_HashTable* ht = hashInit(struct Mystruct);
for (i = 0 ; i < numsSize ; i++) {
mystruct[t].key = nums[i];
struct Mystruct* ret = hashSafeInsertField(ht, mystruct[t], key);
if (ret) {
ret->num++;
if (ret->num > numsSize >> 1) {
free(mystruct);
hashDestroy(ht);
return nums[i];
}
}
else
mystruct[t++].num = 1; // 注意初始为1,而不是0
}
return -1; // 此处不应被执行到
}
在Discuss里看到了一个更好的解决方案,9ms,O(1)空间,叹为观止!!
int majorityElement(int num[], int n) {
int cnt = 0, res, i;
for (i = 0; i < n; ++i) {
if (cnt == 0) res = num[i];
if (res == num[i]) {
if (++cnt > n >> 1)
return res;
}
else --cnt;
}
return res;
}
res保存结果,cnt计数领先值。
这种方法必须最终返回res,而不是判断cnt > n>>1即可返回。因为对于[3, 4, 3]的情况,可能3已经出现了超过一半的次数,但它的领先值(cnt)依然为0。判断cnt > n>>1只是一个小小的优化。
而且这种方法判断是否存在一个Majority Element,但是幸亏题目指出一定存在一个Majority Element(这也正是面试的关键所在)。否则还需O(n)时间和空间判断Majority Element的存在。