用fortran编写迭代法解三对角矩阵

以下是用Fortran编写的三对角矩阵的迭代法求解代码：

program tridiagonal
    implicit none
    integer, parameter :: n = 5 ! 矩阵的大小
    real :: a(n), b(n), c(n), f(n) ! 矩阵及方程组右端项
    real :: alpha(n), beta(n), x(n) ! 存储中间变量和解
    integer :: i
    ! 初始化矩阵和右端项
    do i = 1, n-1
        a(i) = -1.0
        b(i) = 2.0
        c(i) = -1.0
        f(i) = 1.0
    end do
    b(n) = 2.0
    f(n) = 1.0
    ! 初始化中间变量
    alpha(1) = b(1)
    beta(1) = f(1) / alpha(1)
    ! 迭代计算中间变量
    do i = 2, n
        alpha(i) = b(i) - a(i)*c(i-1) / alpha(i-1)
        beta(i) = (f(i) - a(i)*beta(i-1)) / alpha(i)
    end do
    ! 计算解
    x(n) = beta(n)
    do i = n-1, 1, -1
        x(i) = beta(i) - c(i)*x(i+1) / alpha(i)
    end do
    ! 输出结果
    do i = 1, n
        write(*,*) "x(",i,") = ", x(i)
    end do
end program tridiagonal

其中 `a(i)`, `b(i)`, `c(i)` 分别表示矩阵第 $i$ 行的下对角线元素、主对角线元素和上对角线元素，`f(i)` 表示方程组的右端项。`alpha(i)` 和 `beta(i)` 是中间变量，`x(i)` 存储解。在迭代计算中间变量时，注意需要按照顺序计算。在计算解时，需要从后往前计算。