求解线性方程组--追赶法
/* 求解线性方程组--追赶法 */
//####################################
//# 本分解法仅适用于当 #
//# 系数矩阵为三对角形阵的情况 #
//####################################
#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;
}