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

Условие

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

Решение

  1. motherland = {}
  2. for i in range(int(input())):
  3.     country, *cities = input().split()
  4.     for city in cities:
  5.          motherland[city] = country
  6. for i in range(int(input())):
  7.     print(motherland[input()])