202006-2 CCF认证考试 Python版 稀疏向量

第一题线性分类器

这个题思路很简单但博主至今没有找到Python100分的答案，（想到办法后再更新也希望有小伙伴找到后可以分享一下）

我尝试了数组和字典两种方法，都运行超时，数组的30分，字典的60分，代码如下：

# 使用数组，运行超时30分
n,a,b=input().split()
n,a,b=int(n),int(a),int(b)
value_1=[]
value_2=[]
index_1=[]
index_2=[]
re=0
for i in range(a):
	i_1,v_1=input().split()
	i_1,v_1=int(i_1),int(v_1)
	index_1.append(i_1)
	value_1.append(v_1)
for j in range(b):
	i_2,v_2=input().split()
	i_2,v_2=int(i_2),int(v_2)
	index_2.append(i_2)
	value_2.append(v_2)
for k in range(a):
	for p in range(b):
		if index_1[k]==index_2[p]:
			re=re+value_1[k]*value_2[p]
print(re)
#使用字典，运行超时60分
n,a,b=input().split()
a,b=int(a),int(b)
dic_1={}
dic_2={}
re=0
for i in range(a):
	key_1,value_1=input().split()
	dic_1[key_1]=value_1
for j in range(b):
	key_2,value_2=input().split()
	if key_2 in dic_1:
		re=re+int(dic_1[key_2])*int(value_2)
print(re)

更新一位大佬的满分答案（好像只有七十分了，评论区有位大佬的答案是满分的，大家可以参考）：

def input2num():
    return map(int, input().split())


import sys


dim, cap_a, cap_b = input2num()
a = {}
tol = 0

for line in sys.stdin.readlines():
    i, n = map( int, line.split() )
    if cap_a > 0:
        a[i] = n
        cap_a -= 1
    else:
        if i in a:
            tol += n*a[i]

    
print(tol)