题解 | 数据分类处理

I = input().split()[1:]
R = input().split()[1:]
r = list(map(str, sorted(set(map(int, R)))))
res = []
lenRes = 0
for rule in r:
    res1 = []
    for id, data in enumerate(I):
        if rule in data:
            res1.append(id)
            res1.append(int(data))
    if res1:
        res.append([rule, len(res1)//2] + list(map(int, res1)))
        lenRes += (len(res1) + 2)
print(lenRes, end = " ")
for re in res:
    if re[1] > 0:
        print(" ".join(map(str, re)), end = " ")