力扣——前K個高頻元素
阿新 • • 發佈:2019-04-06
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 = newPriorityQueue<>(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個高頻元素