Convert the jewels string into a set for O(1) lookups, then iterate through each character in stones and count how many are in the jewels set.
sum()Use a generator expression with sum() to count stones that appear in the jewels set. Same logic as Solution 1, but written more concisely.
""" 0771.1 - Solution 1 - Hash Set Lookup """
#####################################################################################
# Imports
#####################################################################################
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def numJewelsInStones(self, jewels: str, stones: str) -> int:
"""Jewels and Stones Function"""
jewel_set = set(jewels)
count = 0
for stone in stones:
if stone in jewel_set:
count += 1
return count
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().numJewelsInStones("aA", "aAAbbbb")) # 3
print(Solution().numJewelsInStones("z", "ZZ")) # 0
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 0771.2 - Solution 2 - One-Liner with sum() """
#####################################################################################
# Imports
#####################################################################################
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def numJewelsInStones(self, jewels: str, stones: str) -> int:
"""Jewels and Stones Function"""
return sum(s in set(jewels) for s in stones)
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().numJewelsInStones("aA", "aAAbbbb")) # 3
print(Solution().numJewelsInStones("z", "ZZ")) # 0
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()