解题方法:摩尔投票法
摩尔投票法分为两个阶段,一个是投票阶段,一个是计数阶段,投票阶段用于进行票数对抗,筛选出可能符合某个条件的候选列表,计数阶段进行票数验证。
最初的摩尔投票法可用于筛选出超过一般票数的候选人,在线性时间线性复杂度的情况下实现。
首先定义一个候选人与候选票数,对投票列表进行遍历,如果当前遍历的人与候选人不同,此时如果候选票数为0,则更换候选人为当前遍历的人,候选票数更新为1,如果候选票数大于1,则候选票数减一。如果当前遍历的人与候选人相同,则候选票数加一。最后如果得到候选人的票数大于0,则进行计数阶段,统计该候选人的票数是否满足条件,否则,没有符合条件的候选人。
根据这个基础的摩尔投票法可以引申出,判断超过n/3,n/4等条件的候选人。
判断超过n/3的候选人的代码如下。
vector<int> majorityElement(vector<int>& nums) {
pair<int, int> p1, p2;
p1 = make_pair(nums[0], 0);
p1 = make_pair(nums[0], 0);
for (int i = 0; i < nums.size(); i++) {
if (nums[i] == p1.first) {
p1.second++;
} else if (nums[i] == p2.first) {
p2.second++;
} else {
if (p1.second == 0 && p2.second > 0) {
p1.first = nums[i];
p1.second = 1;
} else if (p2.second == 0 && p1.second > 0) {
p2.first = nums[i];
p2.second = 1;
} else if (p1.second > 0 && p2.second > 0){
p1.second--;
p2.second--;
} else if (p1.second == 0 && p2.second == 0) {
p1.first = nums[i];
p1.second = 1;
}
}
}
int count1 = 0, count2 = 0;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] == p1.first)
count1++;
else if (nums[i] == p2.first)
count2++;
}
vector<int> rtn;
if (count1 * 3 > nums.size())
rtn.push_back(p1.first);
if (count2 * 3 > nums.size())
rtn.push_back(p2.first);
return rtn;
}