摩尔投票算法 和 N/K 推广
算法简介
博耶-摩尔多数投票算法(英语:Boyer–Moore majority vote algorithm),中文常作多数投票算法、摩尔投票算法等,是一种用来寻找一组元素中占多数元素的常数空间级时间复杂度算法。
算法求解思路
- 创建一个数组来存储候选数和计数器(词频),计数器初始化为 0。
- 然后对每个数进行遍历操作,在遍历到元素 x 的时候
- 如果对应的计数器为 0,则将 x 存储到对应的位置中去,并将计数器设置为 1。
- 否则,将 x 与存储的元素进行比较,然后递增计数器(如果它们相等)或递减计数器(不相等)。
- 在此过程结束时,如果数组中存在占多数的元素,则检验每个数是否满足条件
代码实现
注意: 使用前提是存在多数元素,否则需要最后来检验
169. 多数元素
求解k=2时,寻找一个元素
>
n
2
> \frac{n}{2}
>2n,用两个变量表示即可,代码如下:
int majorityElement(vector<int>& nums) {
int candidate = -1;
int count = 0;
for(int num : nums){
if(num == candidate)count++;
else if(--count<0){
candidate = num;
count = 1;
}
}
cout << candidate;
return candidate;
}
N K \frac{N}{K} KN扩展:
思路:
- 遍历每个数
- 若候选容器有该数,词频+1
- 若没有该数,且侯选容器没满(即词频为0),则添加该候选数
- 若没有该数,且候选容器满了,则每个候选数词频-1
- 检验剩余的候选数是否满足词频数 > n k > \frac{n}{k} >kn
vector<int> majorityElementCommon(vector<int> nums, int k){
int n = nums.size();
vector<vector<int>> candidate(k,vector<int>(2,0));
auto update = [&](int x){
for(int i=0;i<k-1;i++){
//发现candidate中有该数
if(x == candidate[i][0]){
candidate[i][1]++;
return;
}
}
//candidate中没有该数
for(int i=0;i<k-1;i++){
//candidate没有满
if(candidate[i][1]==0){
candidate[i][0] = x;
candidate[i][1] = 1;
return;
}
}
//全部抵消
for(int i=0;i<k-1;i++){
candidate[i][1]--;
}
return;
};
for(int num : nums){
update(num);
}
auto check = [&](int x){
return ranges::count(nums,x)>n/k;
};
//剩余的数进行验证即可
vector<int> res;
for(int i=0;i<k-1;i++){
cout<<candidate[i][0] << " ";
if(candidate[i][1] && check(candidate[i][0])){
res.push_back(candidate[i][0]);
}
}
return res;
}