【桶排序、堆排序】Leetcode --- NO.451 根据字符出现的频率排序（Java）

题目描述

题目分析

• 这道题就是找出每个字符出现的频率，然后根据频率来重组字符串
• 频率多的放前面，频率少的放后面，频率一样先后顺序随便，但是要字母要放在一次
• 跟 《Leetcode — NO.347 前K个高频元素》一样的解法
• 这里涉及到 PriorityQueue 类的使用，参考《玩转数据结构之优先队列和堆》

解法分析

• 首先要计算每个字符出现的频率，可以通过 Map 来存储字符和频率的对应关系
• 因为最后要将频率出现的多的字符放在前面，那就要对频率做一个排序
• 这里可以使用桶排序：第 i 个桶中出现的字符就说明出现了 i 次
• 也使用堆排序，跟第 347 那道题大同小异

代码

• 桶排序

class Solution {
    public String frequencySort(String s) {
        // 累计各个字符出现的频率
        Map<Character, Integer> freq = new HashMap<>();
        for(char c : s.toCharArray()){
            freq.put(c,freq.getOrDefault(c, 0)+1);
        }
        // 放到桶中 第 i 个 桶中存储的数据出现了 i 次
        List<Character>[] bucket = new ArrayList[s.length()+1];
        for(char c : freq.keySet()){
            int f = freq.get(c);
            if(bucket[f] == null){
                bucket[f] = new ArrayList<>();
            }
            bucket[f].add(c);
        }
        // 构造结果字符串
        StringBuilder str = new StringBuilder();
        // 从后往前遍历桶，出现频率多的要放到前面
        for(int i = bucket.length-1;i>=0;i--){
            if(bucket[i] == null) continue;
            // 对每个桶中的字母进行遍历
            for(char c : bucket[i]){
                // 第 i 个桶中的字母要加 i 次
                for(int j = 0;j<i;j++){
                    str.append(c);
                }
            }
        }
        return str.toString();
    }
}

• 堆解法

class Solution {
    public String frequencySort(String s) {
        Map<Character, Integer> map = new HashMap<>();
        for (char chr : s.toCharArray()) {
            map.put(chr, map.getOrDefault(chr, 0) + 1);
        }

        PriorityQueue<Map.Entry<Character, Integer>> maxHeap = new PriorityQueue<>(
            (e1, e2) -> e2.getValue() - e1.getValue());

        maxHeap.addAll(map.entrySet());

        StringBuilder sortedString = new StringBuilder(s.length());
        while (!maxHeap.isEmpty()) {
            Map.Entry<Character, Integer> entry = maxHeap.poll();
            for (int i = 0; i < entry.getValue(); i++){
                sortedString.append(entry.getKey());
            }
        }
        return sortedString.toString();
    }
}

作者：donoghl
链接：https://leetcode-cn.com/problems/sort-characters-by-frequency/solution/java-top-k-5-by-donoghl-2/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。