347.前 K 个高频元素

发布时间:2024年01月19日

使用map和sort

class Dual{
    int key;
    int value;
    Dual(int key,int value){
        this.key=key;
        this.value=value;
    }
}
class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        int res = 0;
        for (int i = 0; i < nums.length; i++) {
            if (!map.containsKey(nums[i])) {
                map.put(nums[i], 0);
                res++;
            } else {
                map.put(nums[i], map.get(nums[i]) + 1);
            }
        }
        Dual[] array = new Dual[res];
        int idx=0;
        for(Map.Entry<Integer, Integer> entry:map.entrySet()){
            array[idx++]= new Dual(entry.getKey(),entry.getValue());
        }
        Arrays.sort(array,(d1,d2)->Integer.compare(d2.value,d1.value));
        int[] result = new int[k];
        for(int i=0;i<k;i++){
            result[i]=array[i].key;
        }
        return result;
    }
}

首先用map记录每个value出现的频次key,然后倒入数组中,以key进行sort排序,然后再返回前k个value

使用小顶堆

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        int res = 0;
        for (int i = 0; i < nums.length; i++) {
            if (!map.containsKey(nums[i])) {
                map.put(nums[i], 0);
                res++;
            } else {
                map.put(nums[i], map.get(nums[i]) + 1);
            }
        }
        PriorityQueue<int[]> pq = new PriorityQueue<>((pair1, pair2) -> pair1[1] - pair2[1]);
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (pq.size() < k) {
                pq.offer(new int[] { entry.getKey(), entry.getValue() });
            } else {
                if (entry.getValue() > pq.peek()[1]) {
                    pq.poll();
                    pq.offer(new int[] { entry.getKey(), entry.getValue() });
                }
            }
        }
        int[] result = new int[k];
        for (int i = 0; i < k; i++) {
            result[i] = pq.poll()[0];
        }
        return result;
    }
}
文章来源:https://blog.csdn.net/qq_44047415/article/details/135701711
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。