追赶法解三对角矩阵fortran

以下是追赶法解三对角矩阵的 Fortran 代码示例：

SUBROUTINE TRIDIAG_SOLVER(N, A, B, C, R, X)
C
C   N: 矩阵维度
C   A, B, C: 矩阵对角线系数
C   R: 矩阵右侧向量
C   X: 解向量
C
      INTEGER N
      REAL A(N), B(N), C(N), R(N), X(N)
C
C   前向追赶
      DO 10 I = 2, N
        A(I) = A(I) / (B(I) - A(I-1) * C(I-1))
        C(I) = C(I) / (B(I) - A(I-1) * C(I-1))
        B(I) = B(I) - A(I-1) * C(I-1)
        R(I) = R(I) - A(I-1) * R(I-1)
  10  CONTINUE
C
C   后向追赶
      X(N) = R(N) / B(N)
      DO 20 I = N-1, 1, -1
        X(I) = (R(I) - C(I) * X(I+1)) / B(I)
  20  CONTINUE
C
      RETURN
      END

这段代码实现了一个名为 `TRIDIAG_SOLVER` 的子程序，用于解决一个三对角矩阵的线性方程组。参数 `N` 表示矩阵的维度，`A`、`B`、`C` 分别表示矩阵的三条对角线上的系数，`R` 表示矩阵的右侧向量，`X` 表示解向量。在子程序中，首先进行前向追赶，计算出矩阵的新系数和右侧向量，然后进行后向追赶，计算出解向量。