问题描述
给定两个数组，计算它们的点积。
示例
输入 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 |