Contest 4 task 3

#Вам дан словарь, состоящий из пар слов. Каждое слово является синонимом к парному ему слову. Все слова в словаре различны. Для одного данного слова определите его синоним.
#Программа получает на вход количество пар синонимов N. Далее следует N строк, каждая строка содержит ровно два слова-синонима. После этого следует одно слово.
#Программа должна вывести синоним к данному слову. Примечание
#Эту задачу можно решить и без словарей (сохранив все входные данные в списке), но решение со словарем будет более простым.

digit = int(input())
dictionary = {}
for i in range(digit):
    first_word, second_word = list(map(str, input().split()))
    dictionary[first_word] = second_word

word = input()
for key, value in dictionary.items():
    if word == value:
        print(key)
    elif word == key:
        print(value)