追赶法比较特殊,比前面的算法实现起来都要简单,唯一要注意的是gamma的坐标从1开始
程序分别用3个数组存放三条对角线
代码如下:
///对于系数矩阵是三对角矩阵的方程组可以用追赶法
#include <iostream>
#include <cstdio>
#include <math.h>
using namespace std;
const int MAXN = 1000;
double a[MAXN];
double d[MAXN];
double c[MAXN];
double b[MAXN];
double A[MAXN];
double B[MAXN];
double x[MAXN];
double y[MAXN];
int main()
{
int n;
freopen("in.txt","r",stdin);
while(cin >> n){
for(int i = 0;i<n;i++)cin >> b[i];
for(int i = 0;i<n;i++)cin >> a[i];///先输入主对角线
for(int i = 0;i<n-1;i++)cin >> c[i];///再输入上对角线---瞎编的名称
for(int i = 1;i<n;i++)cin >> d[i];///下对角线---瞎编的名称,注意下角标从1开始
///Crout分解
A[0] = a[0];B[0] = c[0]/A[0];
for(int i = 1;i<n;i++){
A[i] = a[i] - d[i]*B[i-1];