LintCode : 主元素
题目
给定一个整型数组,找出主元素,它在数组中的出现次数严格大于数组元素个数的二分之一。
样例
给出数组[1,1,1,1,2,2,2],返回 1
思路
拥有主元素数组的中位数一定是主元素(可以结合数学知识想想,恩),所以这道题的问题就变为求解中位数。求解中位数我们可以采用快排中patition的方法求解。调用patition之后我们得到主元的位置i,如果i在中点左边,那么我们就将右半部分做patition,如果i在中点右边,那么我们就将左半部分做patition直到我们求得位置i在中点位置上,那么我们就可以return nums[i]了。
代码
int partition(vector<int> &A, int p, int r) {
int x = A[r];
int i = p - 1;
if(p < r)
for(int j = p; j < r; j++)
if(A[j] <= x) swap(A[++i], A[j]);
swap(A[i + 1], A[r]);
return i + 1;
}
int majorityNumber(vector<int> nums) {
int p = 0, r = nums.size() - 1;
int len = nums.size();
while(p < r) {
int q = partition(nums, p, r);
if(q == len / 2)
return nums[q];
else if(q < len / 2)
p = q + 1;
else
r = q - 1;
}
}