宝石与石头

Posted 2021-02-10 zbligang

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

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

示例 1:

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

示例 2:

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

 

 

 

解答：

# O(n)的实现
def numJewelsInStones(J, S):
    stores = {}
    for c in S:
        if c in stores:
            stores[c] = stores[c] + 1
        else:
            stores[c] = 1
    print(stores)
    count = 0
    for c in J:
        if c in stores:
            count += stores[c]
    print(count)


J = "aA"
S = "aAAbbbb"
numJewelsInStones(J, S)
J = "z"
S = "ZZ"
numJewelsInStones(J, S)