题目
Given an array of size n, find the majority element. The majority
element is the element that appears more than ⌊ n/2 ⌋ times. You may
assume that the array is non-empty and the majority element always
exist in the array.Example 1: Input: [3,2,3] Output: 3
Example 2: Input: [2,2,1,1,1,2,2] Output: 2
这道题题意很简单,找出一个数组中出现次数大于数组长度一半的数。
一道难度为Easy的题,好像没有需要博客记录的必要,但是这道题的一个巧妙的解法还是值得学习。
先说最容易想到的办法:
1. 解法一
对每个数出现次数都记录下来,形成哈希表,然后遍历一遍哈希表。这是最容易想到的做法,时间复杂度为O(n),但是空间复杂度就很大了。
class Solution {
public:
int majorityElement(vector<int>& nums) {
unordered_map<int, int>mp;
for (auto i : nums) {
if (!mp.count(i)) {
mp[i] = 1;
}
else mp[i]++;
}
for (auto it = mp.begin(); it != mp.end(); it++) {
if (it->second > nums.size() / 2)
return it->first;
}
return 0;
}
};
2. 解法二
解法二的思想是记录出现最多次数的元素soldier以及它的次数cnt,首先不妨认为数组第一个元素就是我们要找的元素,然后遍历数组,每碰到一个元素,如果和记录的soldier相同,就把cnt加1,表示这个元素出现多了一次增大了可能性,如果和记录的soldier不同,就说明数组中有其他元素可能次数比它还多,就把cnt减1,表示减轻它的权重,当减到0的时候就不要减了,因为这时候表示这个元素的重要性和新来的元素已经一致,所以直接替换soldier为新来的元素,这样遍历完,我们记录的soldier就是要找的元素。
到这并不意味着程序就结束了,因为如果这个数组中不存在这样的元素,比如这样的一个数组:1,3,2,4, 由于每个数字都只出现了一次,所以后面的数字会抵消它前面一个数字,那么遍历完数组得到的soldier就是4,可是4并不满足出现的次数大于2,所以这要求在得到soldier之后验证它满足条件。当然这道题目不需要下面的步骤,因为它保证了这个元素的存在。
代码如下:
class Solution {
public:
int majorityElement(vector<int>& nums) {
if (nums.size() == 1) return nums[0];
int soilder = nums[0], cnt = 1;
for (int i =1; i < nums.size(); i++) {
if (nums[i] == soilder) cnt++;
else {
if (cnt > 0) cnt--; //如果当前soilder出现不止一次 cnt抵消一次
else soilder = nums[i]; //如果当前soilder次数被抵消完了,就不再减cnt的值,而是更新soilder的值为新值
}
}
// 检查是否满足条件
cnt = 0;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] == soilder) cnt++;
}
if (cnt > nums.size() / 2) return soilder;
return 0;
}
};
时间复杂度为O(n),空间复杂度相比方法一就降低到了O(1)