Pat甲级1002 in Python

题目链接：
https://www.patest.cn/contests/pat-a-practise/1002

</pre><pre name="code" class="python">from collections import defaultdict
from functools import reduce

'''
数据结构：
num是个defaultdict(list) 用来存放指数和相应的值，通过defaultdict的特性，可以实现value的累加
shanchu是个数组，用来存放指数 which if 指数的value变成零，则删除这个指数
keys 是个数组，用来对num里的key进行逆序排序，然后再让num里的key按照逆序存到n3里
n3 就是最后的输出数组
'''
n1=input().split()
n2=input().split()
n1=list(map(float,n1))
n2=list(map(float,n2))
num = defaultdict(list)
shanchu=[]
for i in range(1,2*int(n1[0])+1)[0::2] :
    num[int(n1[i])].append(n1[i+1])
for j in range(1,2*int(n2[0])+1)[0::2]:
    num[int(n2[j])].append(n2[j + 1])
for key in num.keys():
    num[key]=reduce(lambda x,y:x+y,num[key])
    if num[key]==0:
        shanchu.append(key)   #吧要删除的key先存起来，为什么不在这里直接pop呢，因为pop之后改变了num.keys（）的迭代变量，会报错
for i in shanchu:
    num.pop(i)
#print(num)

n3=[]
n3.append(len(num))
keys=[]
for key in num.keys():
    keys.append(key)
keys.sort(reverse=True)     #先把keys放到数组中进行排序，要不然num里面key的排列是无序的。
for key in keys:
    n3.append(key)
    n3.append(round(num[key],1))
print(n3[0],end='')
for i in n3[1::]:
    print('',i,end='')