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

题目描述

我们来定义一个函数 f(s)，其中传入参数 s 是一个非空字符串；该函数的功能是统计 s 中（按字典序比较）最小字母的出现频次。

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

现在，给你两个字符串数组待查表 queries 和词汇表 words，请你返回一个整数数组 answer 作为答案，其中每个 answer[i] 是满足 f(queries[i]) < f(W) 的词的数目，W 是词汇表 words 中的词。

示例 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] 都是小写英文字母

解法

由题目可知，f 函数是指最小元素的出现频次，要返回的是 queries 数组中每个元素 queries[i] 对应的 W 的个数，W 是指 words 数组中的元素，其元素的 f 函数值大于对应的 queries[i] 的 f 函数值。

最简单的方式自然是遍历 queries 数组，对于每个元素遍历 words 数组，计算出 W 的个数。不过该方式的时间开销较大，不推荐该方式。

因为每个元素的 f 函数值不大于 10，即值的范围是有限的，因此不妨申请固定长度的数组，以 f 值对应数组下标，存储每个 f 值对应的 W 元素的个数。因此只需要 queries 数组的一次遍历，即可获得要返回的结果。

class Solution:
    def numSmallerByFrequency(self, queries: List[str], words: List[str]) -> List[int]:
        arr,count=[0]*11,0
        for word in words:
            arr[word.count(min(word))]+=1
        for i in range(10,0,-1):
            arr[i],count=count,count+arr[i]
        return [arr[e.count(min(e))] for e in queries]