题解 | 矩阵和向量的点积

矩阵和向量的点积

https://www.nowcoder.com/practice/a7c9e4eb3b3745cf8d81d6159051937d

def matrix_vector_dot_product(matrix, vector):
    # 补全代码
    result = -1
    m = len(matrix) #行
    n = len(matrix[0]) #列
    if n == len(vector):
        result = [0]*m
        for i in range(m):
            s = 0
            for j in range(n):
                s += matrix[i][j]*vector[j]
            result[i] = s
    return result

# 主程序
if __name__ == "__main__":
    # 输入矩阵和向量
    matrix_input = input()
    vector_input = input()

    # 处理输入
    import ast
    matrix = ast.literal_eval(matrix_input)
    vector = ast.literal_eval(vector_input)

    # 调用函数计算点积
    output = matrix_vector_dot_product(matrix, vector)
    
    # 输出结果
    print(output)

全部评论

相关推荐

今天 16:04
已编辑
安徽农业大学 算法工程师
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务