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

发布时间 2023-06-10 16:43:35作者: Owlwu

定义一个函数 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] 都由小写英文字母组成

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/compare-strings-by-frequency-of-the-smallest-character
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


 

由于数据的长度较小,所以直接用计数排序来表示所有 words 中的元素对应 f 后的结果。

再通过前缀和,将每一种长度对应 words 中长度比他大的元素进行判断和存储。

最后遍历 queries 即可。

class Solution {
    public int[] numSmallerByFrequency (String[] queries, String[] words) {
        int[] count = new int[11];
        // 计数排序(大概)
        for (String str : words) {
            int tem = f(str);
            count[tem] ++;
        }
        int num = 0;
        int len = words.length;
        // 所有比当前长度大的元素的数量
        for (int i = 0; i < 10; i++) {
            count[i] = len - num;
            if (count[i + 1] != 0) {
                num += count[i + 1];
            }
        }
        int[] res = new int[queries.length];
        for (int i = 0; i < queries.length; i++) {
            int index = f(queries[i]);
            res[i] = count[index];
        }
        return res;
    }

    private int f (String str) {
        int cnt = 0;
        int ch = 'z';
        for (int i = 0; i < str.length(); i++) {
            char c = str.charAt(i);
            if (c < ch) {
                cnt = 1;
                ch = c;
            } else if (c == ch) {
                cnt ++;
            }
        }
        return cnt;
    }
}