[algorithm] 백준 17219 - 비밀번호 찾기 (파이썬)

📚 문제

입력

출력

예제 입력

16 4
noj.am IU
acmicpc.net UAENA
startlink.io THEKINGOD
google.com ZEZE
nate.com VOICEMAIL
naver.com REDQUEEN
daum.net MODERNTIMES
utube.com BLACKOUT
zum.com LASTFANTASY
dreamwiz.com RAINDROP
hanyang.ac.kr SOMEDAY
dhlottery.co.kr BOO
duksoo.hs.kr HAVANA
hanyang-u.ms.kr OBLIVIATE
yd.es.kr LOVEATTACK
mcc.hanyang.ac.kr ADREAMER
startlink.io
acmicpc.net
noj.am
mcc.hanyang.ac.kr

예제 출력

THEKINGOD
UAENA
IU
ADREAMER

🧑🏻‍💻 풀이 과정

• 빠른 입력을 받기 위해 sys.stdin.readline().rstrip().split()을 사용하자.
• n만큼 반복하며 사이트 주소(address)와 비밀번호(pwd)을 입력받아 딕셔너리(_dic)에 저장 후 반환하자.
• m만큼 반복하며 key(사이트 주소)에 해당하는 value(비밀번호)를 출력하자.

import sys


def init_dit(n):
    _dic = {}
    for _ in range(n):
        address, pwd = sys.stdin.readline().rstrip().split()
        _dic[address] = _dic.setdefault(address, pwd)
    return _dic


def find_password(_dic, m):
    for _ in range(m):
        print(_dic[input()])


n, m = map(int, sys.stdin.readline().rstrip().split())
site = init_dit(n)
find_password(site, m)