给定一个非空的整数数组,返回其中出现频率前 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 是数组的大小。
------
思路:
首先利用hashMap,通过遍历一边,存储元素和相应的次数 。 hashMap<元素,次数>
其次,利用一个数组,长度为n+1,数组下标对应出现的次数, 遍历hashmap,将出现 i次的元素,存储在 数组[i]的位置上,感觉类似于倒排索引。
最后,倒序查找这个数组,取k个,即为结果。
public List<Integer> topKFrequent(int[] nums, int k) { if (nums == null || nums.length == 0) { return null; } HashMap<Integer, Integer> hashMap = new HashMap<>(); for (int i = 0; i < nums.length; i++) { if (hashMap.containsKey(nums[i])) { hashMap.put(nums[i], hashMap.get(nums[i]) + 1); } else { hashMap.put(nums[i], 1); } }
ArrayList<Integer>[] list = new ArrayList[nums.length + 1]; for (int key : hashMap.keySet()) { int value = hashMap.get(key); //以value 为下标,将key存储 list[] 数组, list[] 中是一个 arrayList if (list[value] == null) { ArrayList<Integer> temp = new ArrayList<>(); temp.add(key); list[value] = temp; } else { list[value].add(key); } } ArrayList<Integer> res = new ArrayList<>(); //存放结果 for (int i = list.length - 1; i >= 0 && res.size() < k; i--) { if (list[i] != null) { res.addAll(list[i]); } } return res; }