HJ25. 数据分类处理

97 阅读1分钟

1. 题目

www.nowcoder.com/practice/9a…

image.png

2. 考点

读题读题。

3. 核心代码

def inp():
    l = input().split(' ')[1:]
    r = input().split(' ')[1:]
    r = sorted(set([int(i) for i in r]))
    result = {}
    for j in r:
        for i, v in enumerate(l):
            if str(j) not in v: continue
            result.setdefault(j, []).append([i, int(v)])
    ret = []
    for r in result:
        ret.append(r)
        ret.append(len(result[r]))
        for i in result[r]:
            ret.extend(i)
    ret.insert(0, len(ret))
    return ' '.join([str(i) for i in ret])


if __name__ == '__main__':
    print(inp())