LintCode ：主元素

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;
    }
}