[algorithm] leetcode 771 - Jewels and Stones (파이썬)

문제

J는 보석이며, S는 갖고 있는 돌이다.

S에는 보석이 몇 개나 있을까?

₩대소문자는 구분₩한다.

leetcode 771 - Jewels and Stones

코드

class Solution:
    def numJewelsInStones(self, J: str, S: str) -> int:
        freqs = {}
        count = 0

        # 돌(S)의 빈도 수 계산
        for char in S:
            # 처음이면 1 저장
            if char not in freqs:
                freqs[char] = 1
            else:
            # 이미 있으면 +1
                freqs[char] += 1

        # 보석(J)의 빈도 수 합산
        for char in J:
            if char in freqs:
                count += freqs[char]

        return count

조금 더 파이썬스러운 코드

class Solution:
    def numJewelsInStones(self, J: str, S: str) -> int:
        # 파이썬 컴프리헨션
        return sum(s in J for s in S)