内积意思指两个向量每个维度相乘之和。两维度都非零,相乘才有意义。于是找两个向量都非零的维度相乘相加即可。
第一个向量u的键值对存入映射表,第二个向量的键值对就从映射表里找存在的idx,相乘,相加。
更新的代码,现在改用sys.stdin.readlines()代替input读取数据就能100了。
import sys
n, a, b = map(int, input().split())
d, ans = {
}, 0
for line in sys.stdin.readlines():
idx, val = map(int, line.split())
if a > 0:
d[idx] = val
a -= 1
elif idx in d:
ans += d[idx]