1170. 比较字符串最小字母出现频次

题目描述

定义一个函数 f(s)，统计 s 中 （按字典序比较）最小字母的出现频次 ，其中 s 是一个非空字符串。

例如，若 s = "dcce"，那么 f(s) = 2，因为字典序最小字母是 "c"，它出现了 2 次。

现在，给你两个字符串数组：待查表 queries 和词汇表 words 。对于每次查询 queries[i] ，需统计 words 中满足 f(queries[i]) < f(W) 的 词的数目 ，W 表示词汇表 words 中的每个词。

请你返回一个整数数组 answer 作为答案，其中每个 answer[i] 是第 i 次查询的结果。

示例 1：

输入：queries = [“cbd”], words = [“zaaaz”]
输出：[1]
解释：查询 f(“cbd”) = 1，而 f(“zaaaz”) = 3 所以 f(“cbd”) < f(“zaaaz”)。

示例 2：

输入：queries = [“bbb”,“cc”], words = [“a”,“aa”,“aaa”,“aaaa”]
输出：[1,2]
解释：第一个查询 f(“bbb”) < f(“aaaa”)，第二个查询 f(“aaa”) 和 f(“aaaa”) 都 > f(“cc”)。

提示：

1 <= queries.length <= 2000
1 <= words.length <= 2000
1 <= queries[i].length, words[i].length <= 10
queries[i][j]、words[i][j] 都由小写英文字母组成

解法一：排序 + 二分查找

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43


func numSmallerByFrequency(queries []string, words []string) []int {
    arr := make([]int, len(words))
    f := func(word *string) int {
        str := *word
        cnt := 0
        var minCh rune
        for _, ch := range str {
            if cnt == 0 {
                minCh = ch
                cnt++
            } else if ch < minCh {
                minCh = ch
                cnt = 1
            } else if ch == minCh {
                cnt++
            } else {
                // ch > minCh，no operation
            }
        }
        return cnt
    }
    for i := 0; i < len(words); i++ {
        arr[i] = f(&words[i])
    }
    sort.Ints(arr)
    retArr := make([]int, len(queries))
    for i := 0; i < len(queries); i++ {
        cnt := f(&queries[i])
        fmt.Println(cnt, arr[0])
        // 查找第一个大于 cnt 的元素的位置。
        left, right := 0, len(arr)-1
        for left <= right {
            mid := left + (right-left)/2
            if arr[mid] > cnt {
                right = mid - 1
            } else {
                left = mid + 1
            }
        }
        retArr[i] = len(arr) - left
    }
    return retArr
}

解法二：后缀和

官方题解如下：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28


func f(s string) int {
    cnt := 0
    ch := 'z'
    for _, c := range s {
        if c < ch {
        ch = c
        cnt = 1
        } else if c == ch {
        cnt++
        }
    }
    return cnt
}

func numSmallerByFrequency(queries []string, words []string) []int {
    count := make([]int, 12)
    for _, s := range words {
        count[f(s)] += 1
    }
    for i := 9; i >= 1; i-- {
        count[i] += count[i + 1]
    }
    res := make([]int, len(queries))
    for i, s := range queries {
        res[i] = count[f(s) + 1]
    }
    return res
}