Rayman的绝顶之路——Leetcode每日一题打卡15

Leetcode347. 前 K 个高频元素 题目：

给定一个非空的整数数组，返回其中出现频率前 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 个高频元素的集合是唯一的。
• 你可以按任意顺序返回答案。

思路：

1. 当 k = 1 时，只需要用哈希表维护元素出现频率，每一步更新最高频元素即可；
2. 当 k > 1 时，需要使用优先队列；
3. 首先使用HashMap建立一个元素值对应出现频率的哈希表，需要手工填值（ 这个步骤需要 O(N) 时间，其中 N 是列表中元素个数）。
4. 第二步建立堆，堆中添加一个元素的复杂度是 O(log(k))，要进行 N 次复杂度是 O(N)。
5. 最后输出结果。

Java AC代码：

import java.util.*;

public class Leetcode347 {
    static int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap();
        for (int n : nums) {
            map.put(n, map.getOrDefault(n, 0) + 1);
        }

        PriorityQueue<Integer> heap = new PriorityQueue<>((n1, n2) -> map.get(n1) - map.get(n2));

        for (int n : map.keySet()) {
            heap.add(n);
            if (heap.size() > k)
                heap.poll();
        }

        List<Integer> top_k = new LinkedList();
        while (!heap.isEmpty())
            top_k.add(heap.poll());
        Collections.reverse(top_k);
        int[] res = new int[top_k.size()];
        for (int i = 0; i < top_k.size(); ++i) {
            res[i] = top_k.get(i);
        }
        return res;
    }

    public static void main(String[] args) {
        System.out.println(topKFrequent(new int[]{1, 1, 1, 2, 2, 3}, 2));
    }
}

2020.4.29打卡