【LeetCode】宝石与石头的python实现

题目：宝石与石头

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

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

示例1

输入: J = “aA”, S = “aAAbbbb”
输出: 3

示例2

输入: J = “z”, S = “ZZ”
输出: 0

写法1：使用for循环和if语句

class Solution:     
	def numJewelsInStones(self, J: str, S: str) -> int:
		count = 0
		for i in S:
			if i in J:
				count += 1
		return(count)
J = "aA"
S = "aAAbbb"
print(Solution().numJewelsInStones(J,S))

写法2：使用列表生成式使得代码更加紧凑

class Solution:     
	def numJewelsInStones(self, J: str, S: str) -> int:
		count = [ i in J for i in S].count(True)
		return(count)
J = "aA"
S = "aAAbbb"
print(Solution().numJewelsInStones(J,S))

输出结果：

3
[Finished in 0.1s]

更新：
count一句还可这么写：

count = sum(i in J for i in S)

或者这么写：

count = sum(S.count(x) for x in J)