202006-2 稀疏向量

試題編號： 202006-2
試題名稱： 稀疏向量
時間限制： 2.0s
記憶體限制： 512.0MB
語言：python

使用陣列卡在60分，使用字典通過。

Code:

import sys
dimension, x_nums, y_nums = map(lambda x:int(x), input().split())
dict_x = {}
result = 0
for line in sys.stdin.readlines():
    location, value = map(lambda x: int(x), line.split())
    if x_nums > 0:
        dict_x[location] = value
        x_nums -= 1
    else:
        if location in dict_x:
            result += dict_x[location] * value
print(result)

Test Case

10 3 4
4 5
7 -3
10 1
1 10
4 20
5 30
7 40