[백준/파이썬] 17219번 비밀번호 찾기

-코드

n, m = map(int, input().split())
site_password = {}
for _ in range(n):
    site, password = input().split()
    site_password[site] = password
for _ in range(m):
    print(site_password[input()])

처음에는이런 코드로 만들었습니다.

n, m = map(int, input().split())
arr = []
for _ in range(n):
    site, password = input().split()
    arr.append([site, password])
for _ in range(n):
    find_password = input()
    for i in arr:
        if find_password == i[0]:
            print(i[1])
            break

하지만 입출력은 맞게 나오는데 틀려서 바꿔보고 해도 해결이 안 되어서 다른 방법을 찾아서 해결을 했습니다.