追赶法求解三对角方程组

#include <stdio.h>
#include <math.h>
#include <stdlib.h>

void read(int a, int b, double arr[]);			//依次读入 arr[a], arr[a + 1], ... , arr[b]
void Doolittle(int n, double d[], double e[], double f[], double l[], double r[]);	//对方阵 A 做 Doolittle 分解
int Chase(int n, double b[], double f[], double l[], double r[], double x[]);		//追赶过程求解,返回值提示是否求解成功

int main(){
	
	int n;
	double b[105], d[105], e[105], f[105], l[105],  r[105];
	double x[105];
	int flag;
	
	scanf("%d", &n);
	read(2, n, d);
	read(1, n, e);
	read(1, n - 1, f);
	read(1, n, b);
	
	Doolittle(n, d, e, f, l, r);
	flag = Chase(n, b, f, l, r, x);
	
	if(flag){
		for(int i = 1; i <= n; i++){
			printf("%lf ", x[i]);
		}
	}
	else{
		printf("Fail.");
	}
	
} 

void read(int a, int b, double arr[]){
	
	for(int i = a; i <= b; i++){					//读入 arr[a], arr[a + 1], ... , arr[b]
		scanf("%lf", &arr[i]);
	}
	
}

void Doolittle(int n, double d[], double e[], double f[], double l[], double r[]){
	
	r[1] = e[1];
	for(int i = 2; i <= n; i++){		//Doolittle分解
		l[i] = d[i] / r[i-1];
		r[i] = e[i] - l[i] * f[i-1];
	}
	
}

int Chase(int n, double b[], double f[], double l[], double r[], double x[]){
	double y[105];
	y[1] = b[1];						//追过程,求解 Ly=b
	for(int i = 2; i <= n; i++){
		y[i] = b[i] - l[i] * y[i-1];
	}
	
	x[n] = y[n] / r[n];						//赶过程,求解 Ux=y
	for(int i = n - 1; i >= 1; i--){	//若有无法求解的情况,返回 0 值
		x[i] = (y[i] - f[i] * x[i+1]) / r[i];
	}
  
	return 1;
	
}

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值