该题默认存在占比多于一半的元素,故该元素的数量 - 其他元素的数量 > 0,可用摩尔投票法。
思路:
- 候选元素(cur)初始化为nums[0],票数cnt初始化为1。
- 当遇到与cur相同的数,则票数cnt++,否则票数cnt- -。
- 当票数cnt为0时,更换候选元素,并将票数cnt重置为1。
int majorityElement(vector<int>& nums) {
int cnt = 1;
int cur = nums[0];
for(int i = 1; i < nums.size(); i++){
if(cur == nums[i]){
cnt++;
}else{
cnt--;
if(cnt == 0){
cur = nums[i+1];//更换候选元素
}
}
}
return cur;
}
而现在这题则可能不存在占比多于一半的元素,若还是使用上述的方法,会存在下面两种问题:
- 当数组为 [2,2,1,1] 时,遍历到最后一个元素,cnt 此时为0,更换候选元素,则导致越界。基于此,我们加上越界判断。
- 当数组为 [2,2,1,1,3] 时,此时不存在占比多于一半的元素。遍历到倒数第二个元素时,cnt 此时为0,更换候选元素,cur = 3,导致最后返回元素为3,而不是-1。基于此,我们判断最后得到的候选元素出现的频率即可。
int majorityElement(vector<int>& nums) {
map<int, int> mp;
for(int num : nums){
mp[num]++;
}
int cur = nums[0];
int cnt = 1;
for(int i = 1; i < nums.size(); i++){
if(cur == nums[i]){
cnt++;
}else{
cnt--;
if(cnt == 0){
if(i + 1 == nums.size()){
return -1;//越界则说明遍历完了,cnt却为0,不存在主要元素
}
cur = nums[i+1];
}
}
}
//若存在多数元素,此时的cur一定是多数元素
return mp[cur] > nums.size()/2 ? cur : -1;
}