DefaultDict Tutorial

  • + 0 comments

    from collections import defaultdict import sys input = sys.stdin.read data = input().splitlines()

    n, m = map(int, data[0].split()) group_a = data[1:n+1]

    group_b = data[n+1:]

    word_positions = defaultdict(list)

    for idx, word in enumerate(group_a, 1): word_positions[word].append(str(idx))

    for word in group_b: if word in word_positions: print(" ".join(word_positions[word])) else: print("-1")