Thomas 求解三对角矩阵 （追赶法）v2 （简化版 0开始）

2019独角兽企业重金招聘Python工程师标准>>>

Thomas 求解三对角矩阵 （追赶法） （简化版 0开始）

减少了中间变量，数组从零开始，用指针传入

int TM0(double *a,double *b,double *c,double *r,double *x,int n)
{
	//原矩阵形式 要求 对角占优 a[0]=0 c[n]=0
	//b0 c0
	//a1 b1 c1
	//   a2 b2 c2
	//    ···
	//         an bn
	double *g,*h;
	g=new double [n+1];
	h=new double [n+1];
	g[0]=c[0]/b[0];
	h[0]=r[0]/b[0];
	for(int i=1;i<=n;i++){
		g[i]=c[i]/(b[i]-a[i]*g[i-1]);
		h[i]=(r[i]-a[i]*h[i-1])/(b[i]-a[i]*g[i-1]);
	}
	x[n]=h[n];
	for(int i=n-1;i>=0;i--){
		x[i]=h[i]-g[i]*x[i+1];
	}
	delete[] g;
	delete[] h;
	return 0;
}

转载于:https://my.oschina.net/u/929434/blog/98623