TreeMap按value排序

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

给定一个字符串数组，再给定整数，请返回出现次数前名的字符串和对应的次数。

返回的答案应该按字符串出现频率由高到低排序。如果不同的字符串有相同出现频率，按字典序排序。

对于两个字符串，大小关系取决于两个字符串从左到右第一个不同字符的ASCII值的大小关系。

比如"ah1x"小于"ahb"，"231"<”32“。字符仅包含数字和字母。

数据范围：字符串数满足 ，每个字符串长度 ，

要求：空间复杂度，时间复杂度

思路：将字符串数组存入TreeMap。TreeMap默认是按照key进行排序，如果需要按照value进行排序的话，就需要将其转换为List，然后利用Collections.sort()进行排序，这样还可以进行双关键字排序。TreeMap转化为List的方式如下：

List<Map.Entry<String,Integer>> list = new ArrayList<>(map.entrySet());

Java代码如下：

import java.util.*;


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
        TreeMap<String,Integer> map = new TreeMap<>();
        for(String s : strings) {
            map.put(s,map.getOrDefault(s,0) + 1);
        }
        List<Map.Entry<String,Integer>> list = new ArrayList<>(map.entrySet());
        Collections.sort(list,new Comparator<Map.Entry<String,Integer>>() {
            public int compare(Map.Entry<String,Integer> e1,Map.Entry<String,Integer> e2) {
                if(e1.getValue() == e2.getValue()) {
                    return e1.getKey().compareTo(e2.getKey());
                } else {
                    return e2.getValue() - e1.getValue();
                }
            }
        });
//         for(int i = 0;i < list.size();i++) {
//             System.out.println(list.get(i).getKey() + " " + list.get(i).getValue());
//         }
        String[][] res = new String[k][2];
        for(int i = 0;i < k;i++) {
            res[i][0] = list.get(i).getKey();
            res[i][1] = String.valueOf(list.get(i).getValue());
        }
        return res;
    }
}