向量点积计算

题意

求两个 $n$ 维向量的点积 $\sum a_i b_i$ 。

思路

zip(first, second) 依次配对两个向量的对应分量，生成器表达式计算每对乘积，sum 完成累加。

代码

Python代码

python

1
2
3
4dimension = int(input())
first = map(int, input().split())
second = map(int, input().split())
print(sum(left * right for left, right in zip(first, second)))

C++代码

cpp

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21#include <cstdio>
int n;
int a[1005];
int b[1005];

int main(){
    scanf("%d",&n);
    int i;
    for (i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for (i=1;i<=n;i++){
        scanf("%d",&b[i]);
    }
    int sum = 0;
    for (i=1;i<=n;i++){
        sum += a[i]*b[i];
    }
    printf("%d\n",sum);
    return 0;
}

复杂度

时间复杂度为 $O(n)$ ，除输入序列外额外空间复杂度为 $O(1)$ 。

总结

对应位置的乘积和是 zip 与生成器表达式的典型组合。

目录

题意

思路

代码

Python代码

C++代码

复杂度

总结