LintCode-46. 主元素

题目描述

给定一个整型数组，找出主元素，它在数组中的出现次数严格大于数组元素个数的二分之一。
注意事项
You may assume that the array is non-empty and the majority number always exist in the array.
样例
给出数组[1,1,1,1,2,2,2]，返回 1
挑战
要求时间复杂度为O(n)，空间复杂度为O(1)

分析

这道题既然已经假定主元素存在并且根据主元素的定义就可以知道这个数组只有两种数字。所以我们定义一个变量temp=0，让它当遇到和第一个数相同的数就加一，不同的就减一。最后根据temp的值来判断是哪个数。

代码

class Solution {
public:
    /*
     * @param nums: a list of integers
     * @return: find a  majority number
     */
    int majorityNumber(vector<int> &nums) {
        // write your code here
        int temp = 0, index;
        for(int i = 0; i < nums.size(); i++) {
            if(nums[i] == nums.front()) {
                temp++;
            } else {
                temp--;
                index = i;
            }
        }
        if(temp > 0) return nums.front();
        else return nums[index];
    }
};