BOJ 17219. 비밀번호 찾기 (Python)

문제

BOJ 17219

 

내가 생각한 풀이

비밀번호를 찾으려고 하는 사이트의 주소를 입력하면 비밀번호가 출력되는 프로그램.

 

딕셔너리의 키에 주소를 주고 값에 비밀번호를 줘서 찾으면 됨.

딕셔너리의 get은 O(1), 딕셔너리에 N번 추가하고 M번 확인하기 때문에 O(N + M)의 시간복잡도를 가진다.

 

코드 (Python)

import sys
input = sys.stdin.readline

N, M = map(int, input().split())
d = dict()
for _ in range(N):
    addr, pwd = input().split()
    d.setdefault(addr, pwd)

for _ in range(M):
    addr = input().strip()
    print(d.get(addr))