前k个高频单词；（如果频率一样的话，让首字母小的排在前面） -- 难点比较器的重写

花朝666

于 2020-11-17 19:37:54 发布

阅读量105

点赞数 1

文章标签： java map 比较器

本文链接：https://blog.csdn.net/qq_51339153/article/details/109750060

版权

```public static List<String> topKFrequent(String[] words, int k) {
        Map<String ,Integer> map = new HashMap<>();
        for(String str : words){
            map.put(str,map.getOrDefault(str,0)+1);
        }
        PriorityQueue<String> q = new PriorityQueue<>(new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if(map.get(o1) == map.get(o2)){
                    return o2.compareTo(o1);//虽然这个堆是个小堆;因为堆弹出去后要反转,所以再堆里面让首字母字母小
                    // 的排在字母大的后面
                }else {
                    return map.get(o1) - map.get(o2);
                }
            }
        });
        for(String s : map.keySet()) {
           q.offer(s);
           if(q.size() > k){
               q.poll();
           }
        }
        List<String> list = new ArrayList<>();
        while ( !q.isEmpty()){
            list.add(q.poll());
        }
        Collections.reverse(list);
        return list;
    }