字符串 Leetcode 771 宝石与石头

2019-07-16  本文已影响0人  禾木清清

题目

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

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

示例 1:

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

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/jewels-and-stones
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路一 24ms

使用字典,先建立宝石的字典,然后统计数

代码

class Solution(object):
    def numJewelsInStones(self, J, S):
        """
        :type J: str
        :type S: str
        :rtype: int
        """
        J_dic = {}
        
        for c in J:
            J_dic[c] = 0
            
        for s in S:
            if s in J_dic:
                J_dic[s] += 1
        total = 0
        for key in J_dic:
            total += J_dic[key]
            
        return total

解法二 16ms

使用字符串自带的count函数。这中方法简便的多。

class Solution(object):
    def numJewelsInStones(self, J, S):
        """
        :type J: str
        :type S: str
        :rtype: int
        """
        return sum(S.count(i) for i in J)
上一篇下一篇

猜你喜欢

热点阅读