字符串出现次数的TopK问题

题目描述

给定一个字符串数组，再给定整数k，请返回出现次数前k名的字符串和对应的次数。
返回的答案应该按字符串出现频率由高到低排序。如果不同的字符串有相同出现频率，按字典序排序。
对于两个字符串，大小关系取决于两个字符串从左到右第一个不同字符的 ASCII 值的大小关系。
比如"ah1x"小于"ahb"，“231”<”32“
字符仅包含数字和字母
实例一：
输入：
[“a”,“b”,“c”,“b”],2
返回值：
[[“b”,“2”],[“a”,“1”]]
说明：
"b"出现了2次，记[“b”,“2”]，"a"与"c"各出现1次，但是a字典序在c前面，记[“a”,“1”]，最后返回[[“b”,“2”],[“a”,“1”]]

分析

建立小根堆，字符串的排序要让字典序靠后的排前面，所以是后compareTo前。

代码

public class Solution {
    /**
     * return topK string
     * @param strings string字符串一维数组 strings
     * @param k int整型 the k
     * @return string字符串二维数组
     */
    public String[][] topKstrings (String[] strings, int k) {
        // write code here
        HashMap<String,Integer>sc=new HashMap<>();
        for(String s:strings){
            sc.put(s,sc.getOrDefault(s,0)+1);
        }
        PriorityQueue<String>heap=new PriorityQueue<>(
            //自定义Comparable接口
            (a,b)->sc.get(a).equals(sc.get(b))?b.compareTo(a):sc.get(a)-sc.get(b));
        for(String s:sc.keySet()){
            heap.offer(s);
            //出堆的是“最不满足要求”的元素
            if(heap.size()>k)heap.poll();
        }
        String[][]res=new String[k][2];
        int j=k-1;
        while(!heap.isEmpty()){
            //出堆并逆序放入数组
            String tmp=heap.poll();
            res[j][0]=tmp;
            res[j][1]=sc.get(tmp)+"";
            j--;
        }
        return res;
    }
}