Leetcode - 771. Jewels and Stones
August 12, 2021
문제
jewels는 보석이며 stones는 갖고있는 돌이다. stones중 보석의 개수를 구하라. 단, 대소문자는 구분한다.
Example 1.
Input: jewels = "aA", stones = "aAAbbbb"
Output: 3Example 2.
Input: jewels = "z", stones = "ZZ"
Output: 0나의 풀이
class Solution:
def numJewelsInStones(self, jewels: str, stones: str) -> int:
d = {}
for s in stones:
if s not in d.keys():
d[s] = 1
else:
d[s] += 1
num = 0
for j in jewels:
if j in d.keys():
num += d[j]
return num
먼저 dictionary자료형 변수 d를 선언해주고 stones의 각 문자를 꺼내와서 만약 d에 존재하지 않는 문자라면 키로 추가해주고 값을 1생성한다. 만약 이미 존재하는 문자(키)라면 해당 키에 값을 1올려준다.
이렇게 생성한 d에 이번엔 jewels의 각 문자를 꺼내와 매칭되는 키의 값을 모두 더한뒤 리턴한다.
풀이 : defaultdict
class Solution:
def numJewelsInStones(self, jewels: str, stones: str) -> int:
d = collections.defaultdict(int)
num = 0
for s in stones:
d[s] += 1
for j in jewels:
num += d[j]
return num
앞선 dictionary자료형인 변수 d를 존재하지 않는 키를 자동으로 생성해주는 collections.defaultdict(int)로 선언하여 키 존재 여부를 체크 하지 않는 풀이이다.
풀이 : Counter
class Solution:
def numJewelsInStones(self, jewels: str, stones: str) -> int:
d = collections.Counter(stones)
num = 0
for j in jewels:
num += d[j]
return num
앞서 stone의 각 문자를 계산하는 로직이 존재했는데 이를 collection.Counter(stones)로 자동으로 처리해주었다. 또한 Counter는 존재하지 않는 키를 참조할때 KeyError를 발생시키지 않고 0을 리턴하기 때문에 collections.defaultdict와 동일하게 에러에 대한 예외처리를 할 필요가 없다.
풀이 : Pythonic Way
class Solution:
def numJewelsInStones(self, jewels: str, stones: str) -> int:
return sum(s in jewels for s in stones)
…?
이 풀이를 쉽게 이해 하기 위해서는 각 단계에 존재하는 List comprehension을 출력해 볼 필요가 있다.
>>> [s for s in stones]
['a', 'A', 'A', 'b', 'b', 'b', 'b']여기까지는 각 문자를 하나씩 출력하는 List comprehension이다.
>>> [s in jewels for s in stones]
[True, True, True, False, False, False, False]이제 여기에 비교구문을 추가하여 jewels에 포함된 문자면 True, 아니라면 False를 반환하게 한다.
이 결과를 sum하면 되는데 Python의 sum은 인수로 iterator를 받으므로 List comprehension을 의미하는 대괄호([])는 생략 가능하다.
