求解线性方程组--追赶法

求解线性方程组--追赶法

 

/* 求解线性方程组--追赶法 *///####################################//# 本分解法仅适用于当 #//# 系数矩阵为三对角形阵的情况 #//#####################################include <iostream>using namespace std;/* 二维数组动态分配模板 */template <class T>T** Allocation2D(int m, int n){ T **a; a = new T*[m]; for (int i=0; i<m; i++){a[i] = new T[n];} return a;}int main(){ /* 循环变量 */ int i, j;/* 系数矩阵的行数 */int n;/* 增广矩阵 */ float** a;/* 动态生成增广矩阵 */cout<<"输入系数矩阵的N值，N：";cin>>n;a = Allocation2D<float>(n, n+1);/* 输入增广矩阵的各值 */cout<<endl<<"输入增广矩阵的各值：/n";for(i=0; i<n; i++){for(j=0; j<n+1; j++){cin>>a[i][j];}}/* 追的过程 */a[0][1] = a[0][1] / a[0][0];a[0][n] = a[0][n] / a[0][0];for(i=1; i<n; i++){a[i][i] = a[i][i] - a[i][i-1] * a[i-1][i]; a[i][n] = (a[i][n] - a[i][i-1] * a[i-1][n]) / a[i][i];if(n-1 == i){break;}a[i][i+1] = a[i][i+1] / a[i][i];} /* 赶的过程 */for(i=n-2; i>=0; i--){a[i][n] = a[i][n] - a[i][i+1] * a[i+1][n];} /* 输出过程 */ cout<<"解向量为：/n"; for(i=0; i<n; i++) {cout<<"x"<<i<<": "<<a[i][n]<<endl; } return 0;}