451. 根据字符出现频率排序 中等

给定一个字符串,请将字符串里的字符按照出现的频率降序排列。

示例 1:

输入:
"tree"

输出:
"eert"

解释:
'e'出现两次,'r''t'都只出现一次。
因此'e'必须出现在'r''t'之前。此外,"eetr"也是一个有效的答案。

示例 2:

输入:
"cccaaa"

输出:
"cccaaa"

解释:
'c''a'都出现三次。此外,"aaaccc"也是有效的答案。
注意"cacaca"是不正确的,因为相同的字母必须放在一起。

示例 3:

输入:
"Aabb"

输出:
"bbAa"

解释:
此外,"bbaA"也是一个有效的答案,但"Aabb"是不正确的。
注意'A''a'被认为是两种不同的字符。

代码参考:

package main

import (
    "fmt"
    "sort"
    "strings"
)

func main() {
    fmt.Println(badFrequencySort("tree")) // eetr
    fmt.Println(frequencySort("tree"))    // eetr
}

// heap 数组实现的优先队列左右子节点的优先度不分大小
// 所以需要重新组织优先度的排序
type Counter struct {
    r rune // rune
    p int  // priority // 出现的次数
}

func frequencySort(s string) string {
    r2c := make(map[rune]int)
    for _, r := range s {
        r2c[r]++
    }

    var counters []Counter
    for r, c := range r2c {
        counters = append(counters, Counter{r, c})
    }

    // 自定义优先度排序规则
    sort.Slice(counters, func(i, j int) bool {
        return counters[i].p > counters[j].p
    })

    s = ""
    for _, counter := range counters {
        s += strings.Repeat(string(counter.r), counter.p)
    }
    return s
}

// 最直观的硬核实现
func badFrequencySort(s string) string {
    r2c := make(map[rune]int)
    for _, r := range s {
        r2c[r]++
    }

    c2rs := make(map[int][]rune)
    for r, c := range r2c {
        c2rs[c] = append(c2rs[c], r)
    }

    counts := make([]int, 0, len(c2rs))
    for c := range c2rs {
        counts = append(counts, c)
    }

    sort.Ints(counts)

    var res string
    for i := len(counts) - 1; i >= 0; i-- {
        for _, r := range c2rs[counts[i]] {
            res += strings.Repeat(string(r), counts[i])
        }
    }

    return res
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng