Задача на словари

from io import StringIO
# представим что эту строку мы на самом деле прочитали из файла
text="""2
Russia Moscow Petersburg Novgorod Kaluga
Ukraine Kiev Donetsk Odessa
3
Odessa
Moscow
Novgorod
"""
 
data = {}
f = StringIO(text) # файловый объект в памяти
n = int(f.readline()) # читаем первую строку из файла
lines = [f.readline() for _ in range(n)] # читаем n строк из файла
m = int(f.readline()) # читаем цифру запросов
requests = [f.readline().strip() for _ in range(m)] # читаем m строк запросов
# создаем словарь где ключ => страна,значение ключа => города
for s in lines[0:n]:
    country,*cities = s.split()
    data.update({country:cities})
 
# запросы
for country,cities in data.items():
    for city in requests:
        if city in cities:
            print(country)

for city in requests:
    for country,cities in data.items():
        if city in cities:
            print(country)

data = {}
f = open('input.txt', 'r') 
n = int(f.readline()) 
lines = [f.readline() for _ in range(2)]
m = int(f.readline()) 
requests = [f.readline().strip() for _ in range(m)]
for s in lines[0:n]:
    country,*cities = s.split()
    data.update({country:cities})
for city in requests:
    for country,cities in data.items():
        if city in cities:
            print(country)
f.close()

lines = [f.readline() for _ in range(n)]

A = dict()
D = []
F = []
x = int(input())
for n in range (x):  
    F = input().split()
    key = F[0]
    del F[0]
    val = F
    A[key]=val
word = int(input())
for i in range (word):
    D.append(input())
for i in range(word):
    for key, val in A.items():
        if D[i] in val:
            print(key)

ConT = {}
for i in range(int(input())):
    line = input()
    country = line[:line.find(' ')].strip()
    towns = line[line.find(' ') + 1:].strip()
    ConT[towns] = country
ConT2 = []
for j in range(int(input())):
    country2 = input()
    ConT2.append(country2)
    for key, value in ConT.items():
        if ConT2[j] in key:
            print(value)

# создаем словарь где ключ => город, значение ключа => страна
for s in lines[0:n]:
    country,*cities = s.split()
    for city in cities:
        data.update({city:country})
 
# запросы
for city in requests:
    print(city,data[city])

for s in lines[0:n]:
    country,*cities = s.split()
    data.update(dict.fromkeys(cities,country))
 
print(data) 
# запросы
for city in requests:
    print(city,data[city])