我目前只在编写一个小程序,用C++中的有限差分方法求解微分方程。问题是我正在使用lapack求解我的三对角矩阵,但是在调用例程后却遇到了分段错误。该例程仍以info = 0退出,因此我不知道问题出在哪里。这是程序。
#include <iostream>
#include <cmath>
#include <vector>
#define pi 3.14159265358979323846
using namespace std;
extern "C" void dstev_(char* job, int* N, double* D, double* OFFD, double* EV, int* VDIM, double* WORK, int* INFO);
int main(){
int i = 0;
// systems parameters
double a = 2e10-6;
double k = pi/1.55e-6;
double n1_sq = pow(3.0, 2), n2_sq = pow(3.1, 2);
double step = 1e-6;
double factor = 1/(k*k*step*step);
// parameters for lapack
char job = 'V';
int N = 80, vdim = 100, info;
// create and initialize vector arrays for lapack routine
vector<double> d, offd, work;
vector< vector<double> > ev(vdim, vector<double>(N));
// set up diagonal elements
d.push_back(n1_sq - factor);
for(i=1; i<N-1; i++){
if(i<N/4 || i>=3*N/4){
d.push_back(n1_sq - 2*factor);
}
else{
d.push_back(n2_sq - 2*factor);
}
}
d.push_back(n1_sq - factor);
// set up off diagonal elements
for(i=0; i<N-1; i++){
offd.push_back(factor);
}
// initialize other arrays
for(i=0; i<vdim; i++){
work.push_back(0.0);
}
cout << "Before routine" << endl;
dstev_(&job, &N, &*d.begin(), &*offd.begin(), &*(ev.begin()->begin()), &vdim, &*work.begin(), &info);
cout << "After routine"<< endl;
return 0;
}
这是我第一次使用lapack,所以我不知道此代码有什么不对/对,但我们欢迎您提供任何帮助。另外,为了将 vector 传递到例程中,为什么我们需要传递begin()迭代器?这是我学习的方法,但我认为我并不完全了解它。
最佳答案
我不知道这是否是此代码的唯一问题,但是std::vector< std::vector< double> >
不会分配连续内存...因此,Lapack将无法访问矩阵ev
的所有元素。您应该提供一个n*n
大小的 vector ,矩阵的(i,j)
条目为i+j*n
。
关于c++ - Lapack例行dstev中的段错误,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/42608297/