数组中出现次数超过一半(可实现时间复杂度o(n),空间复杂度o(1))的进阶版
鸽巢原理:出现次数大于N/K的元素的个数至多为(M-1)个
思路:
每次从数组中删除k个不同的元素,直到不能再删了为止。那么最后数组中剩余的元素就是候选元素。
因此 可建立一个k长度的map记录从数组中待消的元素,例如数组a={4,3,3,9,4,2,1,4,3,4,9,2} N=12,k=3 ,则每次应该删3个数,依次遍历数组:
| | | |
| | 3 | 3 |
4 | 4 3 | 4 3 | 4 3 9 (此时map的size为3,删除最后一行439) |
直到遍历完数组。。。。map中留下的元素即为备选元素,再返过去统计备选元素出现次数是否真的是大于N/K
时间复杂度o(kN) 空间复杂度o(K) map的空间
#include <iostream> #include <string> #include <vector> #include <map> #include <algorithm> using namespace std; vector<int> GetNumUpToNK(vector<int> a,int K) { vector<int> res; int len = a.size(); if (len == 0) return res; map<int, int> Ktemp;//待消map K项 for (int i = 0; i < len; ++i) { int atemp = a[i]; if (Ktemp.find(atemp) != Ktemp.end())//该数在map中 { ++Ktemp[atemp]; }else //不在就加入map中,并判断map的size是否为K,为K则消除K个不同数,相应计数-- { Ktemp.insert(make_pair(atemp, 1)); if (Ktemp.size() == K) { for (auto iter = Ktemp.begin(); iter != Ktemp.end();) { --((*iter).second); if ((*iter).second == 0) iter = Ktemp.erase(iter); else ++iter; } } } } //此时Ktemp中的元素为备选元素,继续核实。。。。。。 for (auto it = Ktemp.begin();it!=Ktemp.end();++it) (*it).second = 0; for (auto c:a) //使用此方法c为只读 { if (Ktemp.find(c) != Ktemp.end()) { ++Ktemp[c]; } } for (auto b:Ktemp) { if (b.second>len/K) { res.push_back(b.first); } } return res; } int main() { vector<int> a = { 7,3,3,7,4,5,4,7,3,4,3,4,7,3,4 }; vector<int> ress = GetNumUpToNK(a, 3); if (ress.empty()) { cout << "无符合要求的元素" << endl; } else { for (auto m : ress) cout << m << " "; cout << endl; } }