题解 | #数据分类处理#

非常残暴的解法，但是速度内存比较好看

num = input().split()[1::]
num1 = input().split()[1::]  # 第一个数字对我的作用不大，去掉了
a = 0
for i in num:
    num[a] = [a, i]  # 在所有数据前加上索引
    a += 1
num2 = sorted(set(num1))  # 对第二行数据去重排序
dic = {}
b = 0
for n in num2:
    lis = []
    for m in num:
        if n in m[1]:  # 只保存存在的数据
            b += 1
            lis.append(m)
            dic.setdefault(n, lis)   # 将所有包含第二行元素的第一行元素作为第一行元素的值
coun = (b + len(dic)) * 2    # 就是这个数
lis = []
for answer in dic:
    lis1 = []
    lis1.append(answer)
    lis1.append(len(dic[answer]))
    for answ in dic[answer]:  # 将所有数据读取出来方便输出
        lis1.append(answ[0])
        lis1.append(answ[1])  # 懒得优化了
    lis.append(lis1)
print(coun, end=" ")
lis.sort(key=lambda x:int(x[0]))
for i in lis:
    for j in i:
        print(j, end=" ")  # 输出的最后应该有一个多余的空格，但是也给过了？？

题目逻辑不难，只是数据处理比较繁琐