347. 前 K 个高频元素（java实现）--LeetCode

题目：

给定一个非空的整数数组，返回其中出现频率前 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：heap

/**
 * 思路：
 * 统计词频在新的map上，k是数，v是词频
 * heap中存放词频，统计前k个
 */
    public int[] topKFrequent(int[] nums, int k) {
        int[] res=new int[k];
        HashMap<Integer, Integer> map = new HashMap<>();
        PriorityQueue<Integer> heap = new PriorityQueue<>((v1, v2) -> map.get(v2) - map.get(v1));
        for (int i:nums){
            map.put(i,map.getOrDefault(i, 0)+1);
        }
        for(int key:map.keySet()){
            heap.add(key);
        }
        for (int i=0;i<k;i++){
            res[i]=heap.remove();
        }
        return res;
    }

时间复杂度：Onlogn

空间复杂度：On

解法2：二叉搜索树

/**
 * 思路：
 * 二叉搜索树
 * 统计词频在新的map上，k是数，v是词频
 * TreeMap中存放，k是词频，v是数的集合
 * 之后获取前k高词频的数
 */
    public int[] topKFrequent(int[] nums, int k) {
        int[] result = new int[k];
        HashMap<Integer, Integer> map = new HashMap<>();
        for(int i:nums){
            map.put(i,map.getOrDefault(i,0)+1);
        }
        TreeMap<Integer, List<Integer>> bST = new TreeMap<>();
        for (int num:map.keySet()){
            Integer frequent = map.get(num);
            if (!bST.containsKey(frequent)){
                bST.put(frequent,new ArrayList());
            }
            bST.get(frequent).add(num);
        }
        List<Integer> list = new ArrayList<>();
        while (list.size()<k){
            list.addAll(bST.pollLastEntry().getValue());
        }
        for (int i=0;i<k;i++){
            result[i]= list.get(i);
        }
        return result;
    }

时间复杂度：On

空间复杂度：On