1. 程式人生 > >力扣——前K個高頻元素

力扣——前K個高頻元素

hashmap 返回 add 算法 輸出 get contain 示例 urn

給定一個非空的整數數組,返回其中出現頻率前 k 高的元素。

示例 1:

輸入: nums = [1,1,1,2,2,3], k = 2
輸出: [1,2]

示例 2:

輸入: nums = [1], k = 1
輸出: [1]

說明:

  • 你可以假設給定的 k 總是合理的,且 1 ≤ k ≤ 數組中不相同的元素的個數。
  • 你的算法的時間復雜度必須優於 O(n log n) , n 是數組的大小。
class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        HashMap
<Integer, Integer> map = new HashMap<>(); for(int num : nums){ if(!map.containsKey(num)) { map.put(num, 1); }else { map.put(num, map.get(num) + 1); } } PriorityQueue<Integer> queue = new
PriorityQueue<>(new Comparator<Integer>() { @Override public int compare(Integer arg0, Integer arg1) { return map.get(arg0) - map.get(arg1); } }); for(int key : map.keySet()) { if(queue.size() < k) { queue.add(key); }
else if(map.get(key) > map.get(queue.peek())){ queue.poll(); queue.add(key); } } LinkedList<Integer> linkedList = new LinkedList<>(); while(!queue.isEmpty()) { linkedList.addFirst(queue.poll()); } return linkedList; } }

力扣——前K個高頻元素