TopK问题——统计大家最爱玩的游戏

问题：统计大家最爱玩的前K种游戏

思路：大体思路是用库里的map来存放数据pair< string,int>可以统计。然后再比较int的值建大堆，堆排序之后取出前K的元素，存放在数组里返回。

代码：

//核心代码
#include<algorithm>
#include<map>
#include<string>
#include<vector>
using namespace std;
class Greater
{
public:
bool operator()(pair<string, int>p1, pair<string, int>p2)
{
return p1.second > p2.second;//重载()，注意要比较second的值大小。
}
};
vector<string> FavoriteGamesTopK(vector<string>& game , int K)
{
vector<string>TopKGame;
map<string, int> Countmap;//创建数据结构存储数据
map<string, int>::iterator It;//设置迭代器遍历
for (size_t i = 0; i < game.size(); ++i)
{
//方案一 判断插入
It = Countmap.find(game[i]);
if (It != Countmap.end())
{
//找到
It->second++;
}
else
Countmap.insert(pair<string, int>(game[i],1));
//方案二 用operator[]插入。
//Countmap[game[i]]++;
//operator[]底层实现是用insert，也就是判断有就给sceond++，没有就插入。
}
vector<pair<string,int>> heap;
heap.insert(heap.begin(),Countmap.begin(), Countmap.end());//把Countmap的元素插入到heap中
make_heap(heap.begin(),heap.end(),Greater());//建立大堆 这里默认的缺省值为less，就是小堆，所以自己给出了仿函数Greater（）来实现建立大堆
sort_heap(heap.begin(), heap.end(), Greater());
for (int i = 0; i <K; ++i)//取出前K的元素
{
TopKGame.push_back(heap[i].first);
}
return TopKGame;
}

测试：

void TestGetTopK()
{
vector<string> games;
games.push_back("英雄联盟");
games.push_back("王者荣耀");
games.push_back("王者荣耀");
games.push_back("DNF");
games.push_back("CS:GO");
games.push_back("英雄联盟");
games.push_back("王者荣耀");
games.push_back("DNF");
games.push_back("王者荣耀");
games.push_back("data2");
games.push_back("data2");
games.push_back("魔法王座");
games.push_back("贪玩蓝月");
games.push_back("英雄联盟");
vector<string> ret=FavoriteGamesTopK(games,5);//K=5
for (int i = 0; i < ret.size(); ++i)
{
cout <<"Top "<<i+1<<":"<< ret[i] <<endl ;
}
}
int main()
{
TestGetTopK();
return 0;
}

测试结果：