题解 | 生词篇章查询

from collections import defaultdict
import sys

n, d = int(sys.stdin.readline()), defaultdict(set)
for i in range(1, n + 1):
    s = sys.stdin.readline().split()
    for x in s[1:]:
        d[x].add(i)
m = int(sys.stdin.readline())
for _ in range(m):
    w = sys.stdin.readline().strip()
    print(" ".join(map(str, sorted(list(d[w])))) if w in d else "")