0堆/哈希中等 NC97 字符串出现次数的TopK问题

NC97 字符串出现次数的TopK问题

描述

给定一个字符串数组，再给定整数k，请返回出现次数前k名的字符串和对应的次数。返回的答案应该按字符串出现频率由高到低排序。如果不同的字符串有相同出现频率，按字典序排序。对于两个字符串，大小关系取决于两个字符串从左到右第一个不同字符的 ASCII 值的大小关系。比如"ah1x"小于"ahb"，“231”<”32“。字符仅包含数字和字母

分析

• 使用map集合统计每个字符串的出现的次数
• 使用大顶堆，或者小顶堆给map集合的entry排序，编写比较器
• 输出排序后的entry

import java.util.*;
public class Solution {
    public String[][] topKstrings (String[] strings, int k) {
        if(strings.length == 0 || strings.length < k){
            return null;
        }
        HashMap<String,Integer> map = new HashMap<>();
        for(String str : strings){
            if(map.containsKey(str)){
                map.put(str,map.get(str) + 1);
            }else{
                map.put(str,1);
            }
        }

        PriorityQueue<Map.Entry<String,Integer>> minHeap = new PriorityQueue<>((a,b)->{
            int res =  a.getValue() - b.getValue();
            if(res == 0){
                return b.getKey().compareTo(a.getKey());
            }
            return res;
        });

        for(Map.Entry<String,Integer> entry : map.entrySet()){
            minHeap.add(entry);
        }

        while(minHeap.size() > k){
            minHeap.poll();
        }
        String[][] res = new String[k][2];
        for(int i = k - 1;  i >= 0; i--){
            Map.Entry<String,Integer> entry = minHeap.poll();
            res[i][0] = entry.getKey();
            res[i][1] = ""+entry.getValue();
        }
        return res;
    }
}