692:前K个高频单词

This commit is contained in:
huangge1199 2021-05-20 09:58:51 +08:00
parent 3bf3bb81e0
commit b36f0c69b4
2 changed files with 124 additions and 0 deletions

View File

@ -0,0 +1,83 @@
//给一非空的单词列表返回前 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
//
//
//
//
// 注意
//
//
// 假定 k 总为有效值 1 k 集合元素数
// 输入的单词均由小写字母组成
//
//
//
//
// 扩展练习
//
//
// 尝试以 O(n log k) 时间复杂度和 O(n) 空间复杂度解决
//
// Related Topics 字典树 哈希表
// 👍 265 👎 0
package leetcode.editor.cn;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.PriorityQueue;
//692:前K个高频单词
public class TopKFrequentWords{
public static void main(String[] args) {
//测试代码
Solution solution = new TopKFrequentWords().new Solution();
}
//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public List<String> topKFrequent(String[] words, int k) {
HashMap<String, Integer> map = new HashMap<>();
for (String s : words) {
map.put(s, map.getOrDefault(s, 0) + 1);
}
PriorityQueue<String> queue = new PriorityQueue<>((o1, o2) ->{
if (map.get(o1).equals(map.get(o2))) {
return o1.compareTo(o2);
}
return map.get(o2) - map.get(o1);
});
queue.addAll(map.keySet());
List<String> result = new ArrayList<>();
for (int i = 0; i < k; i++) {
result.add(queue.peek());
queue.poll();
}
return result;
}
}
//leetcode submit region end(Prohibit modification and deletion)
}

View File

@ -0,0 +1,41 @@
<p>给一非空的单词列表,返回前&nbsp;<em>k&nbsp;</em>个出现次数最多的单词。</p>
<p>返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。</p>
<p><strong>示例 1</strong></p>
<pre>
<strong>输入:</strong> [&quot;i&quot;, &quot;love&quot;, &quot;leetcode&quot;, &quot;i&quot;, &quot;love&quot;, &quot;coding&quot;], k = 2
<strong>输出:</strong> [&quot;i&quot;, &quot;love&quot;]
<strong>解析:</strong> &quot;i&quot;&quot;love&quot; 为出现次数最多的两个单词均为2次。
注意,按字母顺序 &quot;i&quot;&quot;love&quot; 之前。
</pre>
<p>&nbsp;</p>
<p><strong>示例 2</strong></p>
<pre>
<strong>输入:</strong> [&quot;the&quot;, &quot;day&quot;, &quot;is&quot;, &quot;sunny&quot;, &quot;the&quot;, &quot;the&quot;, &quot;the&quot;, &quot;sunny&quot;, &quot;is&quot;, &quot;is&quot;], k = 4
<strong>输出:</strong> [&quot;the&quot;, &quot;is&quot;, &quot;sunny&quot;, &quot;day&quot;]
<strong>解析:</strong> &quot;the&quot;, &quot;is&quot;, &quot;sunny&quot;&quot;day&quot; 是出现次数最多的四个单词,
出现次数依次为 4, 3, 2 和 1 次。
</pre>
<p>&nbsp;</p>
<p><strong>注意:</strong></p>
<ol>
<li>假定 <em>k</em> 总为有效值, 1 &le; <em>k</em> &le; 集合元素数。</li>
<li>输入的单词均由小写字母组成。</li>
</ol>
<p>&nbsp;</p>
<p><strong>扩展练习:</strong></p>
<ol>
<li>尝试以&nbsp;<em>O</em>(<em>n</em> log <em>k</em>) 时间复杂度和&nbsp;<em>O</em>(<em>n</em>) 空间复杂度解决。</li>
</ol>
<div><div>Related Topics</div><div><li></li><li>字典树</li><li>哈希表</li></div></div>\n<div><li>👍 265</li><li>👎 0</li></div>