알고리즘 테스트/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을 출력해준다.