追赶法.txt

来自「数值计算方法的一些题」· 文本 代码 · 共 33 行

function X=mypursue(A)
[n,m]=size(A);
a=zeros(1,n);
a(2:n)=diag(A,-1);
c=diag(A,1);
d=diag(A);
if d(1)==0
    'error'
    return
end
p(1)=d(1);
q(1)=c(1)/p(1);
for k=2:n-1
    p(k)=d(k)-a(k)*q(k-1);
    if p(k)==0
        'error'
        return
    end
    q(k)=c(k)/p(k);
end
p(n)=d(n)-a(n)*q(n-1);
    if p(n)==0
        'error'
        return
    end
    y(1)=A(1,n+1)/p(1);
for k=2:n
    y(k)=(A(k,n+1)-a(k)*y(k-1))/p(k);
end
X(n)=y(n);
for k=n-1:-1:1
    X(k)=y(k)-q(k)*X(k+1);
end