169. Majority Element Easy

来源:互联网 发布:java 内存泄露常见场景 编辑:程序博客网 时间:2024/05/16 23:38

这题虽然只是个简单题,但是我觉得还是很值得推敲,主要是考虑怎样写更简短。

一开始我觉得先排序,然后返回第n/2个数就可以了。代码如下:

class Solution {
public:
    int majorityElement(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        return nums[nums.size() / 2];
    }
};

实际上也AC了,但是题目中说的出现次数有向下取整,也就是说,当size为奇数时,有可能majority的出现次数不到一半,这样就行不通了。比如 2 2 2 3 4 5 6,总共七个数,用上面的解法,返回的答案是错误的。不知道为什么这样也能AC,可能是取整的方法有误差?

看讨论有人总结了6种方法,除了上面的那种,其它的都很短,还发现了几个有意思的函数,我自己又写了一个稳妥一点的。代码如下:

class Solution {
public:
    int majorityElement(vector<int>& nums) {
    set<int> s;
        for (int i = 0; i < nums.size(); ++i)
        if(s.find(nums[i]) == s.end()){
        if(count(nums.begin(), nums.begin() + nums.size(), nums[i]) > nums.size() / 2) return nums[i];
        s.insert(nums[i]);
        }
    }
};




0 0
原创粉丝点击