算法---------前 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>() {

            @Override
            public 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 appears
    HashMap<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 heap
    for (int n: count.keySet()) {
      heap.add(n);
      if (heap.size() > k)
        heap.poll();
    }

    // build output list
    List<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 使用：

            @Override
            public 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 数据里面大的，就放进去，把小的出队列。