Given a non-empty array of integers, return the k most frequent elements.
For example,
Given [1,1,1,2,2,3]
and k = 2, return [1,2]
.
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.
题目含义:给出一个非空整数列表,返回出现频次最大的k个元素
1 public List<Integer> topKFrequent(int[] nums, int k) { 2 List<Integer> result = new ArrayList<>(); 3 Map<Integer, Integer> map = new HashMap<>(); 4 for (int i = 0; i < nums.length; i++) { 5 map.put(nums[i], map.getOrDefault(nums[i], 0)+1); 6 } 7 if (map.size() < k) return result; 8 PriorityQueue<Map.Entry<Integer, Integer>> pq = new PriorityQueue<>( 9 new Comparator<Map.Entry<Integer, Integer>>() { 10 @Override 11 public int compare(Map.Entry<Integer, Integer> a, Map.Entry<Integer, Integer> b) { 12 return b.getValue() - a.getValue(); 13 } 14 } 15 ); 16 pq.addAll(map.entrySet()); 17 for (int i = 0; i < k; i++) { 18 result.add(pq.poll().getKey()); 19 } 20 return result; 21 }