题解 | 矩阵和向量的点积

def matrix_vector_dot_product(matrix, vector):
    ans = []
    v_len = len(vector)
    m_width = len(matrix[0])
    if v_len != m_width:
        return -1
    for m in matrix:
        tmp = 0
        for i in range(v_len):
            tmp += m[i] * vector[i]
        ans.append(tmp)
    return ans

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)