前K个高频单词

问题描述 :
给一非空的单词列表，返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率，按字母顺序排序。

示例 1：
输入: [“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2
输出: [“i”, “love”]
解析: “i” 和 “love” 为出现次数最多的两个单词，均为2次。
注意，按字母顺序 “i” 在 “love” 之前。

示例 2：
输入: [“the”, “day”, “is”, “sunny”, “the”, “the”, “the”, “sunny”, “is”, “is”], k = 4
输出: [“the”, “is”, “sunny”, “day”]
解析: “the”, “is”, “sunny” 和 “day” 是出现次数最多的四个单词，
出现次数依次为 4, 3, 2 和 1 次。

输入说明 :
首先输入单词列表的单词数目n，
然后输入n个单词，以空格分隔
最后输入整数k。
假定 k 总为有效值， 1 ≤ k ≤ 集合元素数。
输入的单词均由小写字母组成。

输出说明 :
按题目要求输出，单词之间以一个空格分隔，行首和行尾无多余空格。

输入范例 :
6
i love leetcode i love coding
2

输出范例 :
i love

#include<iostream>
#include<vector>
#include<map>
#include<string.h>
#include<algorithm>
using namespace std;
static bool cmp(pair<string,int> a,pair<string,int> b){
	if(a.second==b.second){
		return a.first<b.first;
	}
	return a.second>b.second;
}
vector<string> topKFrequent(vector<string> &words,int k){
	map<string,int> count;
	for(int i=0;i<words.size();i++){
		count[words[i]]++;
	}
	vector<pair<string,int>> v(count.begin(),count.end());
	sort(v.begin(),v.end(),cmp);
	vector<string> res;
	for(vector<pair<string,int>>::iterator it=v.begin();it!=v.end();it++){
			res.push_back(it->first);
	}
	vector<string> result;
	for(int i=0;i<k;i++){
		result.push_back(res[i]);
	}
	return result;
}
int main(){
	int n;
	string word;
	cin>>n;
	vector<string> words;
	for(int i=0;i<n;i++){
		cin>>word;
		words.push_back(word);
	}
	int k;
	cin>>k;
	vector<string> res=topKFrequent(words,k);
	for(int i=0;i<res.size();i++){
		if(i==res.size()-1){
			cout<<res[i];
		}else{
			cout<<res[i]<<" ";
		}
	}
	return 0;
}