Словари Задача «Страны и города»

Словари

Задача «Страны и города»

Условие

Дан список стран и городов каждой страны. Затем даны названия городов. Для каждого города укажите, в какой стране он находится.

motherland = {}
for i in range(int(input())):
    country, *cities = input().split()
    for city in cities:
        motherland[city] = country
         
for i in range(int(input())):
    print(motherland[input()])

motherland = {}

for i in range(int(input())):

country, *cities = input().split()

for city in cities:

motherland[city] = country

for i in range(int(input())):

print(motherland[input()])