2023.6.23
我的思路是将数组的元素和它出现的频率丢入map容器中。 再用一个for循环,循环k次,每次将最大的key值保存至新数组中,然后将该key值从map中移除,最后即可得到前k个频率最高的元素。 下面上代码:
哈希法:
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int> map;
vector<int> ans;
for(int i=0 ;i<nums.size(); i++)
{
map[nums[i]]++;
}
for(int i=0; i<k; i++)
{
int max = INT_MIN;//比较的频率
int key = 0;//要丢到ans数组的键值
for(auto& it : map)
{
if(it.second > max)
{
max = it.second;
key = it.first;
}
}
ans.push_back(key);
map.erase(key); //移除指定键值的元素
}
return ans;
}
};