追赶法求解三对角线性方程组的MATLAB程序

function [M] = after_method(a,b,c,g,tolerance)

%% 追赶法求解三对角矩阵
% a为三对角矩阵左下对角线上的值
% b为三对角矩阵中间对角线上的值
% c为三对角矩阵右上对角线上的值
% g为方程组右端常数项的值
%%
% M为求解的结果
%%
[m,n] = size(g);
if abs(b(1)) < tolerance
disp('除数为0，停止计算');
return;
else
c(1) = c(1) / b(1);
end
%%
for i = 2:n-1
b(i) = b(i) - a(i-1) * c(i-1);
if abs(b(i)) < tolerance
disp('除数为0，停止计算');
return;
else
c(i) = c(i) / b(i);
end
end
b(n)=b(n)-a(n-1)*c(n-1);
%%
g(1) = g(1) / b(1);
for i = 2:n
g(i) = (g(i) - a(i-1) * g(i-1)) / b(i);
end
%%
M(n) = g(n);
for i=n-1:-1:1
M(i) = g(i) - c(i) * M(i+1);
end
end
%%