leetode 347 前k个高频元素

思路:因为有时间复杂度限制

一 :利用堆排序(nlogn),这里采用java 自带的PriorityQueue队列 (默认是小根堆,所以要重写compartor里的compare()函数,将其改成大根堆,实际上是比较 hashmap频度的大根堆)。总体流程: 先把 num(key) 以及其频度(value) 放入 HashMap中,再将key遍历放入PriorityQueue 中,然后往外取k 个就可以了。

private Map<Integer, Integer> map = new HashMap<>();
    public List<Integer> topKFrequent(int[] nums, int k) {
        for(int num : nums) //把数组按照频度放入
        {
            if(map.containsKey(num))
            {
                map.put(num, map.get(num) + 1);
            }else
            {
                map.put(num, 1);
            }
        }

        PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer a, Integer b) {
                return map.get(b) - map.get(a);
            }
        });
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            pq.add(entry.getKey());
        }
    List<Integer> List = new ArrayList<>();
    for(int i = k; i > 0; i --)
    {
        List.add(pq.poll());
    }
    return List;
    }

二 : 利用桶排序(n + K)

桶排序原理就是设置几个(所提供数组长度为个数)以频度为标准的桶(List)  还是先将num 以及 频度放入Hashmap中,然后遍历Hashmap, 将对应频度的num放入到对应的桶中,接下来从后从后往前遍历桶,选出k个,添加进ansList

public List<Integer> topKFrequent(int[] nums, int k)
{
    Map<Integer, Integer> frequen = new HashMap<>();
    for(int num : nums)
    {
        frequen.put(num, frequen.getOrDefault(num, 0)  + 1);
    }
    List<Integer> [] bucks = new ArrayList[num.lengh + 1];
    for(int key : frequen.keySet())
    {
        int frequency = frequen.get(key);
        if (bucks[frequency] == null)
        {
            bucks[frequency] = new ArrayList<>();
        }
        bucks[frequency].add(key);
    }
    List<Integer> topk = new ArrayList<>();
    for(int i = bucks.lengh - 1; i >= 0 && topk.size() < k; i--)
    {
        if(bucks[i] == null)
        {
            coninue;
        }
        if(bucks[i].size() <= (k - topk.size()))
        {
            topk.addAll(bucks[i]);
        }
        else
        {
            topk.addAll(bucks[i].subList(0, k - topk.size()));
        }
    }
    return topk;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值