追赶法 c++代码实现

肺炎ing，先进制造网课太无聊，写了一份追赶法的代码加强一下理解，h挺low的，仅供参考

#include <iostream>
#include <math.h>
using namespace std;
void trde(int n, double *A, double *b)
{

    int i;
    double u[n];
    double l[n - 1];
    double y[n];
    double x[n];
    //求解L和U矩阵，追的过程
    u[0] = A[0];
    for (i = 0; i < n - 1; i++)
    {
        l[i] = A[2 + 3 * i] / u[i];
        u[i + 1] = A[3 * i + 3] - l[i] * A[1 + 3 * i];
    }
    //求解y矩阵，追的过程 Ly=b
    y[0] = b[0];
    for (i = 0; i < n - 1; i++)
    {
        y[i + 1] = b[i + 1] - l[i] * y[i];
    }
    //求解x矩阵，赶的过程 ux=y
    x[n - 1] = y[n - 1] / u[n - 1];
    for (i = n - 2; i >= 0; i--)
    {
        int c = 0;
        x[i] = (y[i] - x[i + 1] * A[3 * n - 5 - c]) / u[i];
        c += 3;
    }
    for (int i = 0; i < n; i++)
        cout << x[i] << "   ";
}
int main()
{
    int n = 4;         //定义方阵的行列数
    int i = 3 * n - 2; //方阵内非零元素个数
    double A[i] = {-4, 1, 1, -4, 1, 1, -4, 1, 1, -4};//输入方阵的元素
    double b[n] = {3, 4, -4, -3};
    double *d;
    trde(n, A, b);
    cin.get();
    return 0;
}