LintCode 主元素 II

weixin_34006468發表於2017-03-07

題目

給定一個整型陣列,找到主元素,它在陣列中的出現次數嚴格大於陣列元素個數的三分之一。

** 注意事項**
陣列中只有唯一的主元素

樣例
給出陣列[1,2,1,2,1,3,3] 返回 1

分析

三三抵銷法,但是也有需要注意的地方:
我們對cnt1,cnt2減數時,相當於丟棄了3個數字(當前數字,candidate1, candidate2)。也就是說,每一次丟棄數字,我們是丟棄3個不同的數字。
而Majority number超過了1/3所以它最後一定會留下來。
而m > N/3, N - 3x > 0.
3m > N, N > 3x 所以 3m > 3x, m > x 也就是說 m一定沒有被扔完
最壞的情況,Majority number每次都被扔掉了,但它一定會在n1,n2中。

程式碼

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) {
        // write your code
        int candidate1 = 0, candidate2 = 0;
        int count1, count2;
        count1 = count2 = 0;
        
        for(int i=0;i<nums.size();i++) {
            if(candidate1 == nums.get(i)) {
                count1++;
            } else if(candidate2 == nums.get(i)) {
                count2++;
            }else if(count1 == 0) {
                candidate1 = nums.get(i);
                count1 = 1;
            }else if(count2 == 0) {
                candidate2 = nums.get(i);
                count2=1;
            } else {
                count1--;
                count2--;
            }
        }
        
        count1 = count2 = 0;
        for(int i=0;i<nums.size();i++) {
            if(nums.get(i) == candidate1) {
                count1++;
            } else if(nums.get(i) == candidate2) {
                count2++;
            }
        }
        
        return count1>count2?candidate1 :candidate2;
    }
}

相關文章