17219_Find PS - 백준

17219_Find PS

요구사항

• 첫째 줄에 저장된 사이트 주소의 수 N과 비밀번호를 찾으려는 사이트 주소의 수 M이 주어지고, 두번째 줄부터 N개의 줄에 걸쳐 각 줄에 사이트 주소와 비밀번호가 공백으로 구분되어 주어진다.
• N+2번째 줄부터 M개의 줄에 걸쳐 비밀번호를 찾으려는 사이트 주소가 한줄에 하나씩 입력된다. 이때, 반드시 이미 저장된 사이트 주소가 입력되는데, 이것으로 해당 사이트의 비밀번호를 출력하시오.

Solution

from sys import stdin, stdout
# 입력 출력을 가속화 한것이다.
input = stdin.readline
print = stdout.write

N, M = map(int, input().split())
sites = {}

# input으로 받는 site, password를 딕셔너리에 key: value로 저장한다.
for _ in range(N):
    site, ps = input().split(" ")
    sites[site] = ps

# input으로 받는 원하는 site를 키로 넣어 비밀번호를 출력한다.
for _ in range(M):
    print(sites[input().rstrip()])

📌 코드 풀이

1. 전체 사이트와, 찾을 사이트 수를 입력받고 빈 딕셔너리를 선언한다.
2. 전체 사이트만큼 for문을 돌고 site, ps를 공백을 기준으로 입력받고 1번의 딕셔너리에 dic[site] = ps 로 사이트와 비밀번호를 저장해둔다.
3. 찾는 사이트만큼 for문을 돌고, dic[input()]으로 들어오는 사이트를 키값으로 넣어 비밀번호를 반환하면 된다.

백준

https://www.acmicpc.net/problem/17219