[LintCode]主元素

原文发表在我的博客：主元素
求关注、求交流、求意见、求建议。

问题

LintCode：主元素

描述

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

样例

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

挑战

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

实现

这个问题非常简单，不过完成挑战还是需要一个很讨巧的思想，先说一说正常的思路。

排序计数

问题分析

首先想到的是用一个合适的 Tuple 来分别存储每个值的个数，最后取最大的（因为题目默认是有主元素的）。然而这完全是人脑的思路，并不是适合计算机，而且从空间复杂度上就和过挑战无缘了。还是决定先进行排序，再遍历一遍，重复的就计数，大于一半就跳出、返回。

代码实现 - C++

class Solution {
public:
    int majorityNumber(vector<int> nums) {
        sort(nums.begin(), nums.end());
        int len = nums.size();
        int count = 1;
        for(int i = 1; i < len; ++i)
            if(nums[i] == nums[i - 1]){
                count++;
                if(count * 2 > len)
                    return nums[i];
            }else
                count = 1;
        return 0;
    }
};

结果分析

• 结果：23ms 通过测试，速度还不错。
• 分析：由于排序，时间复杂度一定大于 o(n) ，大概为 o(nlogn)，不能完成挑战要求。空间复杂度由于快排占空间复杂度为o(1)，遍历时变量为常数，所以整体空间复杂度为o(1)， 符合挑战要求。

剪裁数组

问题分析

这并不是我想出来的方法，不过这个方法很是巧妙，所以在此记录并拿出来分享一下。核心的思路在于 当一个数组去掉两个不同的元素时，其主元素不改变 。

两个不相等的元素只有两种可能，两种情况分别为 去掉两个非主元素 和 去掉一个主元素和一个非主元素，无论如何主元素比例都不小于去掉之前。

所以从头开始遍历，只要凑成 n 对不相等元素就可以裁剪掉前面的部分。

代码实现 - C++

class Solution {
public:
    int majorityNumber(vector<int> nums) {
        int len = nums.size();
        int count = 0;
        int majority;
        for(int i = 0; i < len; ++i){
            if(count == 0) majority = nums[i];
            if(majority == nums[i]) ++count;
            else --count;
        }
        return majority;
    }
};

结果分析

结果：21ms 通过测试，由于测试数据不大，速度提升并不是很明显。
分析：只遍历一次，时间复杂度o(n)，常数个变量，空间复杂度o(1)，虽然速度提升不明显，但是理论上通过挑战。

总结

简单题想做出花来也并非易事，每道题都有每道题的特点，生搬硬套、死记硬背可以解决很多问题但未必能完美解决。灵活来源于量的积累，多做题，多思考。