每日一题 【每日一题】求点积 -Python-20211006

Jack · 2021年10月06日 · 最后由 lework 回复于 2022年01月21日 · 20 次阅读
  1. 问题描述: 给出两个数组,求他们的点积
  2. 问题示例: A = [1, 1, 1] B = [2, 2, 2] 输出: 6

A = [3, 2] B = [2, 3, 3] 输出: -1 (没有点积)

参考代码:

class Solution:
    def dotProduct(self, 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
package main

import "fmt"

func dotProduct(a []int, b []int) int {
    if len(a) == 0 || len(b) == 0 || len(a) != len(b) {
        return -1
    }
    ans := 0
    for i := 0; i < len(a); i++ {
        ans += a[i] + b[i]
    }
    return ans
}

func main() {
    /*
        求点积
        问题描述: 给出两个数组,求他们的点积
        问题示例: A = [1, 1, 1] B = [2, 2, 2] 输出: 6
        A = [3, 2] B = [2, 3, 3] 输出: -1 (没有点积)
    */
    a := []int{1, 1, 1}
    b := []int{2, 2, 2}
    fmt.Println(dotProduct(a, b))
}
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册