0

我想使用 C++ 中的 Lapack 包求解线性方程组。我计划使用这里的例程，即 dgesv来实现它。

这是我的代码：

unsigned short int n=profentries.size();
double **A, *b, *x;
A= new double*[n];
b=new double[n];
x=new double[2];
// Generate the matrices for my equation
for(int i = 0; i < n; ++i)
{
    A[i] = new double[2];
}

for(int i=0; i<n; i++)
{
    A[i][0]=5;
    A[i][1]=1;
    b[i]=3;
}

std::cout<< "printing matrix A:"<<std::endl;
for(int i=0; i<n; i++)
{
    std::cout<< A[i][0]<<" " <<A[i][1]<<std::endl;
}
// Call the LAPACK solver
    x = new double[n];//probably not necessary
dgesv(A, b, 2, x); //wrong result for x!
std::cout<< "printing vector x:"<<std::endl;

/*prints 
3
3
but that's wrong, the solution is (0.6, 0)!
*/
for(int i=0; i<2; i++) 
{
    std::cout<< x[i]<<std::endl;
}

我有以下问题：

dgesv 怎么会用元素 {3, 3} 计算向量 x？解决方案应该是 {0.6, 0}（用 matlab 检查过）。

问候

编辑： dgesv 可能适用于方阵。我下面的解决方案展示了如何使用 dgels 解决超定系统。

4

2 回答 2

1

Lapack 假设矩阵以列主要格式存储。它不适用于矩阵 A 的指针对指针格式。矩阵的所有元素都需要连续存储在内存中。

尝试这样声明 A：

double *A;
A = new double[2*n];

还有for循环：

for(int i=0; i<n; i++)
{
    A[i+0*n]=5;
    A[i+1*n]=1;
    b[i]=3;
}
于 2012-10-09T10:28:10.390 回答
1

问题曾经是行专业/列专业的混淆，而且显然 dgesv 不适合非方阵。

超定线性方程组可以使用 dgels (#include "lapacke.h") 以最小二乘方式求解。

值得注意的是，对我来说乍一看并不明显：解向量（通常用 x 表示）然后存储在 b 中。

我的示例系统由一个矩阵 A 在其第一列中保存值 1-10 和一个向量 b 组成，其中 {i|1<=i<=10} 的值为 i^2：

    double a[10][2]={1,1,2,1,3,1,4,1,5,1,6,1,7,1,8,1,9,1,10,1};
double b[10][1]={1,4,9,16,25,36,49,64,81,100};
lapack_int info,m,n,lda,ldb,nrhs;
int i,j;

// description here: http://www.netlib.org/lapack/double/dgesv.f
m = 10;
n = 2;
nrhs = 1;
lda = 2;
ldb = 1;

for(int i=0; i<m; i++)
{
    for(int k=0; k<lda; k++)
    {
        std::cout << a[i][k]<<" ";
    }
    std::cout <<"" << std::endl;
}
std::cout<< "printing vector b:"<<std::endl;
for(int i=0; i<m; i++) 
{
    std::cout<< *b[i]<<std::endl;
}
std::cout<< "\nStarting calculation..."<<std::endl;

info = LAPACKE_dgels(LAPACK_ROW_MAJOR,'N',m,n,nrhs,*a,lda,*b,ldb);

std::cout<< "Done."<<std::endl;

std::cout<< " "<<std::endl;
std::cout<< "printing vector b:"<<std::endl;
for(int i=0; i<2; i++) 
{
    std::cout<< *b[i]<<std::endl;
}
std::cout<< "Used values: "<< m << ", " <<  n << ", " << nrhs << ", " << lda << ", " << ldb << std::endl;
于 2012-10-10T11:25:38.477 回答