宝石与石头

宝石与石头

题目

   给定字符串J 代表石头中宝石的类型，和字符串 S代表你拥有的石头。 S 中每个字符代表了一种你拥有的石头的类型，你想知道你拥有的石头中有多少是宝石。
J 中的字母不重复，J 和 S中的所有字符都是字母。字母区分大小写，因此"a"和"A"是不同类型的石头。
示例 1:
输入: J = "aA", S = "aAAbbbb"  输出: 3
示例 2:
输入: J = "z", S = "ZZ"  输出: 0
注意: S 和 J 最多含有50个字母。
J 中的字符不重复。

暴力求解

直接循环石头S，检查每块石头是宝石，时间复杂度为 $O(n^2)$ .

class Solution:
    def numJewelsInStones(self, J: str, S: str) -> int:
        count = 0
        for s in S:
            if s in J:
                count += 1
        return count

使用哈希表求解

将石S头遍历使用哈希表保存每种石头的数目，再遍历宝石J，查看宝石的数目，时间复杂度为 $O(n^2)$ .

def numJewelsInStones(J: str, S: str) -> int:
    s_set = set(S)
    save_dict = dict(zip(s_set, [S.count(s) for s in s_set]))

    return sum([save_dict.get(j, 0) for j in J])