题解 | 数据分类处理

import sys
lst=[]
for line in sys.stdin:
    lst.append(line.split()[1::])
lst_l= lst[0]
lst_R= sorted(map(int,set(lst[1])))
dc={}
ncount=[]
for r in lst_R:
    count_1=0
    for id,l in enumerate(lst_l):
        if str(r) in str(l):
            dc.setdefault(r,[]).append(str(id)+' '+str(l))
            count_1+=1
    ncount.append(count_1)
lenth_1 = len(lst_R)
L=0
k=0
str_ls=[]
for key,val in dc.items():
    k+=1
    str_ls.append(str(key)+' '+str(len(val))+' '+' '.join(val))
    L+=len(val)*2+2
str_2=' '.join(str_ls)
print(str(L)+' '+str_2)