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

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

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

//####################################
//#   本分解法仅适用于当               #
//#   系数矩阵为三对角形阵的情况       #
//####################################

#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;
}