C++ 矩阵*向量乘法中的错误

时间:2021-06-14 14:00:41

标签: c++ loops for-loop nested-loops function-definition

我需要一个将矩阵和向量相乘的函数 (Matrix*vector)

它接受一个矩阵 A 和一个向量 B,用 int 描述维度。不知何故,它没有正确运行。有什么帮助吗??

void Multiply(double *res, double **A, double *B, int ARows, int ACols, int BRows)
{

    if (ACols !=BRows)

    {

        return;
    }
    
    for (int i = 0; i < ACols; i++)
    {
        res[i] = 0;
        for (int j = 0; j < BRows; j++)
        {
            res[i] += A[i][j]*B[j];
        }
    }
}

1 个答案:

答案 0 :(得分:2)

你的意思是

for (int i = 0; i < ARows; i++)
{
    res[i] = 0;
    for (int j = 0; j < ACols; j++)
    {
        res[i] += A[i][j]*B[j];
    }
}

如果函数返回一个布尔值来表示函数执行是否成功会更好

bool Multiply(double *res, double **A, double *B, int ARows, int ACols, int BRows)
{    
   bool success = ACols == BRows;

    if ( success )
    {
        for (int i = 0; i < ARows; i++)
        {
            res[i] = 0;
            for (int j = 0; j < ACols; j++)
            {
               res[i] += A[i][j]*B[j];
            }
        }
    }

    return success;
} 

您可以使用标头 std::inner_product 中声明的标准算法 <numeric>,而不是手动编写的循环。