LeetCode

771. 宝石与石头

2018-11-05  本文已影响1人  闭门造折

给定字符串J 代表石头中宝石的类型,和字符串 S代表你拥有的石头。 S 中每个字符代表了一种你拥有的石头的类型,你想知道你拥有的石头中有多少是宝石。

J 中的字母不重复,J 和 S中的所有字符都是字母。字母区分大小写,因此"a"和"A"是不同类型的石头。

示例 1:

输入: J = "aA", S = "aAAbbbb"
输出: 3

示例 2:

输入: J = "z", S = "ZZ"
输出: 0

注意:
S 和 J 最多含有50个字母。
J 中的字符不重复。

维护一个表,对应a-z,A-Z共52个字符对应位置
如果是宝石,对应位置设置为1

为了更方便映射,我采用了一个辅助函数trans,用来进行字符->数字的翻译操作
遍历一遍J,把tab表里宝石对应索引值置1
遍历一遍S,累加所有的tab表中对应索引值

时间复杂度O(J+S),空间复杂度O(N)(N是选取符号范围)

网上的方法大都是两层for的O(JS)时间复杂度方法
可以减少空间复杂度,这道题规模也比较小,其实没什么区别。

class Solution {
    //翻译函数,字符->数字转换
    public int trans(char c){
        if(c <= 'z' && c >= 'a'){
            return c - 'a';
        }
        return c - 'A' + 30;
    }

    public int numJewelsInStones(String J, String S) {
        //查询用表
        int tab[] = new int[60];
        
        //储存宝石信息
        for(int i = 0; i < J.length(); i++){
            tab[trans(J.charAt(i))] = 1;
        }
        
        //查询手头石头是不是宝石
        int count = 0;
        for(int i = 0; i < S.length(); i++){
            count += tab[trans(S.charAt(i))];
        }
        return count;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读