主元素 II

主元素 II

给定一个整型数组，找到主元素，它在数组中的出现次数严格大于数组元素个数的三分之一。

样例

给出数组[1,2,1,2,1,3,3] 返回 1

注意

数组中只有唯一的主元素

挑战

要求时间复杂度为O(n)，空间复杂度为O(1)。

 

嗯。。 百度了一下。

主元素可能有两个，于是设置两个当前主元素。遍历nums，如果和某个当前住元素相等，则计数加一。如果都不相等

计数减一，若减后计数小于等于0，则将对应的当前住元素更换。

最后需要验证，应为当前主元素只有一个，候选有两个。

 

lintcode上还有个主元素III，想了一下用这种方法也可以做，但是时间复杂度就是O(n*k)了。

public class Solution {
    /**
     * @param nums: A list of integers
     * @return: The majority number that occurs more than 1/3
     */
    public int majorityNumber(ArrayList<Integer> nums) {
        int n = nums.size();
        int max = nums.get(0);
        int max1 = -1;
        int x = 1;
        int x1 = 0;
        for(int i=1;i<n;i++) {
            int m = nums.get(i);
            if(m == max) x++;
            else if(m == max1) x1++;
            else if(x == 0) {
                max = m;
                x = 1;
            }else if(x1 == 0) {
                max1 = m;
                x1 = 1;
            }else {
                x--;
                x1--;
            }
            
            if(3*x > n)return max;
            if(3*x1 > n)return max1;
        }
        x=0;
        x1=0;
        for(int i=0;i<n;i++) {
            if(nums.get(i) == max)x++;
            if(nums.get(i) == max1)x1++;
        }
        if(3*x >n) return max;
        return max1;
    }
}