Given a non-empty array of integers, return the k most frequent elements.
Example 1:
Input: nums = [1,1,1,2,2,3], k = 2
Output: [1,2]
Example 2:
Input: nums = [1], k = 1
Output: [1]
Note:
* You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
* Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

?

共用部分是用map統計頻次。接下來選出topk頻次有variation。

1.O(nlogk), O(k). 堆。
寫專屬comparator，根據Entry<K, V>里的V也就是本題的頻率排序，讓堆頂始終維持是最小頻率的元素，你最菜，當堆大小過k的時候把你擠走哦！

2.O(n), O(n). bucket sort。要用到頻率最多也就是n的信息+O(n)的空間。 第一遍掃計算頻率 第二遍掃根據頻率把元素值放到對應frequency的bucket里。 第三遍掃bucket，從右往左（從高頻到低頻），湊滿k個元素。 細節： 1.bucket因為不能保證某一頻率只有一元素，所以要用List<Integer>[], generic type初始化特別，要List<Integer>[] buckets = new List[nums.length + 1];注意等號右邊不帶generic?

?

實現1：

class Solution {private class EntryComparator implements Comparator<Map.Entry<Integer, Integer>> {@Override// P2: 要過OJ要聲明一下Map.Entry, 沒辦法只寫Entry，因為不夠普遍吧，沒適配。public int compare(Map.Entry<Integer, Integer> a, Map.Entry<Integer, Integer> b) {return a.getValue() - b.getValue();}}public List<Integer> topKFrequent(int[] nums, int k) {Map<Integer, Integer> freqs = new HashMap<>();for (int num : nums) {freqs.put(num, freqs.getOrDefault(num, 0) + 1);}// 1.PQPriorityQueue<Map.Entry<Integer, Integer>> minHeap = new PriorityQueue<>(new EntryComparator());for (Map.Entry<Integer, Integer> e : freqs.entrySet()) {minHeap.offer(e);if (minHeap.size() > k) {minHeap.poll();}}List<Integer> ans = new ArrayList<>();for (int i = 0; i < k; i++) {// P1:注意加到答案的是key不是entry ans.add(minHeap.poll().getKey());}Collections.reverse(ans);return ans;} }

?

?

實現2：

class Solution {public List<Integer> topKFrequent(int[] nums, int k) {Map<Integer, Integer> freqs = new HashMap<>();for (int num : nums) {freqs.put(num, freqs.getOrDefault(num, 0) + 1);}// 1.bucket sort// P1: 注意這個初始化！List<Integer>[] buckets = new List[nums.length + 1];for (int num : freqs.keySet()) {int frequency = freqs.get(num);if (buckets[frequency] == null) {buckets[frequency] = new ArrayList<>();}buckets[frequency].add(num);}List<Integer> ans = new ArrayList<>();for (int i = buckets.length - 1; i >= 0; i--) {if (buckets[i] == null) {continue;}for (int j = 0; j < buckets[i].size() && k > 0; j++) {ans.add(buckets[i].get(j));k--;}}return ans;} }

?

轉載于:https://www.cnblogs.com/jasminemzy/p/9739815.html

《新程序員》：云原生和全面數字化實踐50位技術專家共同創作，文字、視頻、音頻交互閱讀

總結

以上是生活随笔為你收集整理的leetcode347 - Top K Frequent Elements - medium的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。