#include <iostream> #include <vector> #include <unordered_map> #include <queue> #include <algorithm> using namespace std; struct WordFreq { string word; int freq; bool operator<(const WordFreq& other) const { return freq == other.freq ? word > other.word : freq < other.freq; } }; vector<string> topKFrequent(vector<string>& words, int k) { unordered_map<string, int> freq; for (const auto& word : words) { freq[word]++; } priority_queue<WordFreq, vector<WordFreq>> pq; for (const auto& [word, f] : freq) { pq.push({word, f}); if (pq.size() > k) { pq.pop(); } } vector<string> result(k); for (int i = k - 1; i >= 0; i--) { result[i] = pq.top().word; pq.pop(); } return result; } int main() { int k; cin >> k; vector<string> words; string word; while (cin >> word) { if (word == "end") { break; } words.push_back(word); } auto result = topKFrequent(words, k); for (const auto& word : result) { cout << word << " "; } cout << endl; return 0; }

这段代码实现了前 k 个出现次数最多的单词的查找，使用了堆排序的思想，时间复杂度为 O(Nlogk)，其中 N 为单词个数。

具体实现过程如下：

1. 首先使用哈希表统计每个单词出现的频率。

2. 然后使用优先队列（堆）存储单词和对应的频率，队列中最多只保存 k 个元素。

3. 遍历哈希表中的每个元素，将其插入优先队列中，如果队列中元素数量超过 k，则将队列中频率最小的元素弹出。

4. 最后，将优先队列中的 k 个单词按频率从大到小排序，并返回结果。

以下是代码解释：

#include <iostream>
#include <vector>
#include <unordered_map>
#include <queue>
#include <algorithm>

using namespace std;

// 定义一个结构体，用于存储单词和对应的频率
struct WordFreq {
  string word;
  int freq;
  bool operator<(const WordFreq& other) const {
      // 如果频率相同，按字母顺序从小到大排序；否则按频率从大到小排序
      return freq == other.freq ? word < other.word : freq > other.freq;
  }
};

vector<string> topKFrequent(vector<string>& words, int k) {
  // 哈希表用于统计每个单词出现的频率
  unordered_map<string, int> freq;
  for (const auto& word : words) {
      freq[word]++;
  }

  // 优先队列用于存储单词和对应的频率，队列中最多只保存 k 个元素
  priority_queue<WordFreq, vector<WordFreq>> pq;
  for (const auto& [word, f] : freq) {
      pq.push({word, f});
      if (pq.size() > k) {
          pq.pop();
      }
  }

  // 将优先队列中的 k 个单词按频率从大到小排序，并返回结果
  vector<string> result(k);
  for (int i = k - 1; i >= 0; i--) {
      result[i] = pq.top().word;
      pq.pop();
  }
  return result;
}

int main() {
  int k;
  cin >> k;

  vector<string> words;
  string word;
  while (cin >> word) {
      if (word == "end") {
          break;
      }
      words.push_back(word);
  }

  auto result = topKFrequent(words, k);
  for (const auto& word : result) {
      cout << word << " ";
  }
  cout << endl;

  return 0;
}

希望能够帮到你！