【LeetCode-数组】多数元素

题目描述

给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数大于 ⌊ n/2 ⌋ 的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
示例:

输入: [3,2,3]
输出: 3

输入: [2,2,1,1,1,2,2]
输出: 2

题目链接: https://leetcode-cn.com/problems/majority-element/

思路1

由于数组是非空的,并且给定的数组总是存在多数元素。所以,先将数组排序,排序后的数组的第int(n/2)个元素一定是多数元素。代码如下:

class Solution {
public:
    int majorityElement(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        return nums[nums.size()/2];
    }
};
  • 时间复杂度:O(nlogn)
    排序的时间复杂度为O(nlogn)。
  • 空间复杂度:O(1)

思路2

使用投票法。算法如下:

  • 维持两个变量candidata和cnt,candidate表示多数元素的候选,cnt为candidate出现的次数。初始化时candidate可以为任意值,cnt初始化为0;
  • 遍历数组:
    • 如果cnt==0,则将当前元素nums[i]赋值给candidata;
    • 如果当前元素nums[i]与candidate相等,则cnt++;
    • 如果当前元素nums[i]和candidate不等,则cnt--;
  • 循环结束后,candidate就是数组中的多数元素。

代码如下:

class Solution {
public:
    int majorityElement(vector<int>& nums) {
        int candidate = 0;
        int cnt = 0;
        for(int i=0; i<nums.size(); i++){
            if(cnt==0) candidate=nums[i];
            if(candidate==nums[i]) cnt++;
            else cnt--;
        }
        return candidate;
    }
};
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

投票法的简单理解:

  • 如果候选人不是candidate,则candidate会和其他非候选人一起反对候选人,所以候选人一定会下台(cnt==0时发生换届选举)
  • 如果候选人是candidate, 则candidate会支持自己,其他候选人会反对,同样因为candidate票数超过一半,所以candidate一定会成功当选

参考

1、https://leetcode-cn.com/problems/majority-element/solution/duo-shu-yuan-su-by-leetcode-solution/

猜你喜欢

转载自www.cnblogs.com/flix/p/12811218.html
今日推荐