问题描述
给定两个数组,计算它们的点积。
示例
输入 A=[1,2,3], B=[2,3,4] ,输出 20 ;输入 A=[1,2], B=[1,2,3] ,输出 -1 ,即表示没有点积。
代码实现
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
def calculate_dot(a, b): # 数组a, b # 返回点积 if len(a) == 0 or len(b) == 0 or len(a) != len(b): return -1 ans = 0 for i in range(len(a)): ans += a[i] * b[i] return ans A = [1, 2, 3] B = [2, 3, 4] print('输入A:', A) print('输入B:', B) print('输出:', calculate_dot(A, B)) |
运行结果
1 2 3 |
输入A: [1, 2, 3] 输入B: [2, 3, 4] 输出: 20 |