我们来定义一个函数 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] 都是小写英文字母
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/compare-strings-by-frequency-of-the-smallest-character
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
这题难在这个中文翻译贼奇怪,
题目意思是:
返回一个数组res,res[i] 等于 f(queries[i]) 比多少个 f (word for word in words) 小的个数。( f 函数的功能是统计 s 中(按字典序比较)最小字母的出现频次。
比如对于输入:queries = ["bbb","cc"], words = ["a","aa","aaa","aaaa"]
f一下可以得到, f_queries = [3, 2], f_words = [1, 2, 3, 4]
3 只比 4 小,所以res[0] = 1,
2 比 3 和 4 小,所以res[1] = 2。
所以返回 res = [1, 2]。
思路就是,先把f_words一次生成好,然后排好序,
这样每个query都可以直接用二分查找插入的位置,来判断有多少个比 f(query) 大的数。
时间复杂度:O(N + NlogN + M * logN), M是queries的长度, N是words的长度。
空间复杂度:O(N)
class Solution(object):
def numSmallerByFrequency(self, queries, words):
"""
:type queries: List[str]
:type words: List[str]
:rtype: List[int]
"""
def f(word):
return word.count(min(word))
f_words = sorted([f(word) for word in words])
n = len(f_words)
res = []
for q in queries:
cnt = f(q)
res.append(n - bisect.bisect(f_words, cnt))
return res