leet_347_top_k_frequent(前K个频率最大)

• leetcode：地址

• 问题描述：

  • 给定一个非空的数组，输出前K个出现频率最多的数。要求算法的时间复杂度好于O(n lgn)。
  • 输入输出样例：
    • Input: nums = [1,1,1,2,2,3], k = 2
    • Output: [1,2]

• 方法一（大顶堆）分析：

  • 对时间复杂度提出了要求，明显一般的线性操作是达不到的。这里使用的是大顶堆的方式将最大的值进行输出，且每次求解最大值的复杂度为o(lg n)，而具体的实现方式是使用优先队列（priority_queue，是先进最优出，最优一般为最大，其本质上是基于堆的完全二叉树，会自动地对放进队列的元素进行排序）
  • 注意map容器的使用：c++ 数组和一些容器中均会检查下标越界等问题，但map不存在下表越界的概念，这可能与map的实现方式有关（本质上是一个hash表，如除留余数法），且map的下标使用值得注意，不要使用下表来获取对应key的value。原因在于：通过下标索引对map 的key赋值时，如果key存在则将新值覆盖掉旧值；如果索引下表不在map的所有key中，那么就新建一个key,并且将该key对应的value赋默认值（如int 型 默认赋0），然后如果对该key赋新值，则将默认值覆盖掉。

• c++代码：

#include<iostream>
using namespace std;

#include<unordered_map>
#include<vector>
#include<queue>


class Solution 
{
public:
	vector<int> topKFrequent(vector<int>& nums, int k)
	{
		unordered_map<int, int> map;
		for (int num : nums)
		{
			map[num]++;
		}
		
		vector<int> res;
		priority_queue<pair<int, int>> pq;
		for (auto iter = map.begin(); iter != map.end(); iter++)
		{
			pq.push(make_pair(iter->second, iter->first));
			if (pq.size() > (int)map.size() - k)
			{
				res.push_back(pq.top().second);
				pq.pop();
			}
		}
		return res;
	}
	
};

• 方法二（桶排序）

  • 分析：桶排序有些类似哈希表中的链地址法。具体桶排序的原理不在此分析。
  • 代码中的注意点：c++ 标准库中reverse（迭代器1，迭代器2）函数的使用；c++11中开始添加的begin(数组或容器变量),end(数组或容器变量)库函数,其作用与容器的成员函数begin（）、end()相似，只是一个是库函数一个是容器的成员函数。还需注意两种情况下的end()函数始终指向的是尾元素的下一个地址。

  方法二c++代码：

class Solution 
{
public:
	vector<int> topKFrequent_2(vector<int>& nums, int k)
	{
		unordered_map<int, int> counts;
		for (auto i : nums)
			++counts[i];

		vector<vector<int>> buckets(nums.size() + 1);
		for (auto & k : counts)
			buckets[k.second].push_back(k.first);
		reverse(begin(buckets), end(buckets));

		vector<int> res;
		for(auto & bucket:buckets)
			for (auto i : bucket)
			{
				res.push_back(i);
				if (res.size() == k)
					return res;
			}
		return res;
	}
	
};

主调函数：

int main(int argc, char* argv[])
{
	vector<int> num = {2, 2, 2, 3, 1, 9, 5, 1, 4};
	Solution solution = Solution();
	vector<int> res = solution.topKFrequent(num, 2);

	for (int val : res)
	{
		cout << val << endl;
	}
	
	system("pause");
	return 0;
}