CCF 202006-2 稀疏向量 python

题目

感（fei）想(hua)

第一次拿了60分，运行错误，估计是后面4个测试点开销问题。简化解题，AC成功~（撒花✿✿ヽ(°▽°)ノ✿）

忍不住感叹，python大法好，适合我这种编程渣渣~

代码

# 方法1:60分
# 输入
n,a,b=map(int,input().split())
avec={}
bvec={}
for _ in range(a):
	i,v=map(int,input().split())
	avec[i]=v
for _ in range(b):
	i,v=map(int,input().split())
	bvec[i]=v

# 恢复向量
v1=[0]*(n+1)
v2=[0]*(n+1)
for k in avec:
	v1[k]=avec[k]
for k in bvec:
	v2[k]=bvec[k]

# 向量内积
ans=0
for i in range(1,n+1):
	ans+=v1[i]*v2[i]

# 输出
print(ans)

# 方法2：100分
# 输入
n,a,b=map(int,input().split())
avec={}
bvec={}
for _ in range(a):
	i,v=map(int,input().split())
	avec[i]=v
for _ in range(b):
	i,v=map(int,input().split())
	bvec[i]=v

# 直接内积
ans=0
for k in avec:
	if k in bvec:
		ans+=avec[k]*bvec[k]

# 输出结果
print(ans)

Result