알고리즘 테스트/Hackerrank 문제풀이 및 해설
<Hackerrank 문제풀이: 파이썬> DefaultDict Tutorial
개발린이
2020. 11. 28. 19:02
처음에 보았을 때는 무슨 문제인가 잘 이해가 안 갔지만 차근차근 계속 읽어보니 이해가 돼서 간단하게 풀 수 있었다:
from collections import defaultdict
d = defaultdict(list)
n, m = map(int, input().split())
for i in range(1, n+1):
word1 = str(input())
d[word1].append(str(i))
for j in range(m):
word2 = str(input())
if word2 in d:
print(' '.join(d[word2]))
else:
print(-1)
문제 조건에 알맞게 요구하는 글자가 있으면 어디어디 있는지를 출력해주고, 없다면 -1을 출력해준다.