算法---------前 K 个高频元素（Java版本）

本文主要是介绍算法---------前 K 个高频元素（Java版本），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

题目

给定一个非空的整数数组，返回其中出现频率前 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 是数组的大小。
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案。

解决方法

我自己写的：

    public static int[] topKFrequent(int[] nums, int k) {HashMap<Integer,Integer> hashMap = new HashMap<>();for (int num : nums) {int value = hashMap.getOrDefault(num, 0) + 1;hashMap.put(num, value);}int[] result = new int[k];PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(new Comparator<Integer>() {@Overridepublic int compare(Integer o1, Integer o2) {//这里写的太垃圾了 直接 hashMap.get(o1) - hashMap.get(o2) 就行了if (hashMap.get(o1).equals(hashMap.get(o2))) {return 0;}return hashMap.get(o1) > hashMap.get(o2) ? 1 : -1;}});for (Integer integer : hashMap.keySet()) {if (priorityQueue.size() < k) {priorityQueue.add(integer);}else {priorityQueue.add(integer);priorityQueue.poll();}}//这一段真的是垃圾  不能这样遍历队列
//        for (int i = 0; i <= priorityQueue.size(); i++) {
//            result[i] = priorityQueue.poll();
//        }Integer poll ;int i = 0;while ((poll = priorityQueue.poll()) != null){result[i] = poll;i++;}return result;}

网上答案：

class Solution {public List<Integer> topKFrequent(int[] nums, int k) {// build hash map : character and how often it appearsHashMap<Integer, Integer> count = new HashMap();for (int n: nums) {count.put(n, count.getOrDefault(n, 0) + 1);}// init heap 'the less frequent element first'PriorityQueue<Integer> heap =new PriorityQueue<Integer>((n1, n2) -> count.get(n1) - count.get(n2));// keep k top frequent elements in the heapfor (int n: count.keySet()) {heap.add(n);if (heap.size() > k)heap.poll();}// build output listList<Integer> top_k = new LinkedList();while (!heap.isEmpty())top_k.add(heap.poll());Collections.reverse(top_k);return top_k;}
}

学习到的东西

1.PriorityQueue
相对于其他队列，PriorityQueue 是一个有序的队列

2.comparator 使用：

            @Overridepublic int compare(Integer o1, Integer o2) {
//                return o1 - o2;  正序return o2 - o1;   //倒序}

3.int 和 interger 数组的转换
https://caoxinyu.blog.csdn.net/article/details/106406192
4.前k个元素，不需要向冒泡那样麻烦，只需要保存K 个数据，比K 数据里面大的，就放进去，把小的出队列。

这篇关于算法---------前 K 个高频元素（Java版本）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---