CCF CSP 202006-2 稀疏向量 python

202006-2 稀疏向量



这道题思路真的很简单，但是我一直做不对！最多60分了。

两种方法：

使用字典，显示“运行超时”：

data = list(map(int,input().split()))
n,a,b = data[0],data[1],data[2]
out = 0
u = {}
v = {}
 
for i in range(a+b):
    u1,u2 = map(int,input().split())
    if i < a:
        u[u1]=u2
    else:
        if u1 in u :
            out+=u[u1]*u2
print(out)
 

使用列表，显示“运行错误”：

data = list(map(int,input().split()))
n,a,b = data[0],data[1],data[2]
out = 0
u = [0]*n
v = [0]*n
for _ in range(a):
    u1,u2 = map(int,input().split())
    if u2!=0:
        u[u1-1]=u2
for _ in range(b):
    u1,u2 = map(int,input().split())
    if u2!=0:
        v[u1-1]=u2
 
for i in range(n):
    if u[i] != 0 and v[i] != 0:
        out+=u[i]*v[i]
print(out)

如果有大佬可以献上解决方案是最好的了！

更新最新的满分代码：

尝试使用 sys.stdin.readlines()将标准输入全部获取
还有一个要注意的是，如果多加了一个向量i控制读取（如：i < a 和 对应的 i+=1操作），而不是像下面的代码那样的直接在读取到的a上面操作，还是会出现运行超时的问题，下面的代码运行时长为1.906s，好险！

import sys

data = list(map(int,input().split()))
n,a,b = data[0],data[1],data[2]
out = 0
u = {}

# sys.stdin.readlines() 将标准输入全部获取
for line in sys.stdin.readlines():
    u1,u2 = map(int,line.split())
    if a > 0:
        u[u1]=u2
        a -= 1
    else:
        if u1 in u :
            out+=u[u1]*u2

print(out)