[Leetcode] 主元问题

leetcode中有两个题目是讲这个问题的。

Majority Element II

Majority Element

区别是对于主元的个数多少上，一个，两个，三个

这里的陷阱是：当一个元素处于未设置状态的时候，是否要对他进行设置？
如果另一个元素也是未设置状态那么肯定可以设置；如果另一个元素不是未设置状态，那么在这里需要首先监测一下是否和另外那一个元素是相等的，如果是相等，那么不能对这个元素进行设置。

public class Solution {
    public List<Integer> majorityElement(int[] nums) {
        List<Integer> res = new LinkedList<Integer>();
        int count1=0,pre1=0,count2=0,pre2=0;
        for(int i=0;i<nums.length;i++){
            int num = nums[i];
　　　　　　　　　//如果pre1没有设置，需要考虑两种情况
　　　　　　　　　//1. pre2也没有设置 2. pre2虽然设置了，但是和当前值不相等。
            if((count1==0&&(count2==0||(pre2!=num)))||pre1==num){
                pre1=num;
                count1++;
            }else if(count2==0||pre2==num){
                pre2=num;
                count2++;
            }else{
                count1--;
                count2--;
            }
        }
        if(count1!=0){
            int count = 0;
            for(int i=0;i<nums.length;i++){
                if(nums[i]==pre1) count++;
            }
            if(3*count>nums.length)
                res.add(pre1);
        }
        if(count2!=0){
            int count = 0;
            for(int i=0;i<nums.length;i++){
                if(nums[i]==pre2) count++;
            }
            if(3*count>nums.length)
                res.add(pre2);
        }
        return res;
    }
}