thomas2.m

来自「利用MATLAB解决<偏微分方程数值算法>中的一些问题,如插值求解」· M 代码 · 共 18 行

function[U]=Thomas2(A,B,R,D)
n=length(B);
for i=1:n
    for j=1:n
        if i==j
            X(i,j)=B(1,i);
        else if i==j+1
                X(i,j)=A(1,j);
            else if i==j-1
                    X(i,j)=R(1,i);
                else X(i,j)=0;
                end
            end
        end
    end
end
        U=inv(X)*D';