save an algorithm and call it

Question

Nima Vali 2020-10-21

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/621658-save-an-algorithm-and-call-it

评论： Nima Vali 2020-10-21

hello all

I want to save the Thomas algorithm and backward substitution algorithm and call it whenver I need it in my code. How to do this?

solving linear system with Thomas algorithm*******************************
  for k = 1:p-1
i = k +1;
l(i,k)=Ac(i,k)/Ac(k,k);
for j = k :k+1
Ac(i,j) = Ac(i,j)- l(i,k)*Ac(k,j);
end
b(i) = b(i)-l(i,k)*b(k);
  end
   %apply backward substitution
for k = p:-1:1
g(k)=b(k);
for j=k+1: min(p,k+1)
    g(k)=g(k)-Ac(k,j)*g(j);
end
g(k)=g(k)/Ac(k,k);
end
T_cn(2:n-1)=g(1:p);
 end
    

2 个评论
显示无隐藏无

Rik 2020-10-21

Do you mean you want to make this a function?

Nima Vali 2020-10-21

Yes, I want to call it like z=thomas(Ac,b,p-2);

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Rik 2020-10-21

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/621658-save-an-algorithm-and-call-it#answer_520683

在 MATLAB Online 中打开

Store this in thomas.m

function z=thomas(Ac,b,p_minus_2)
%solving linear system with Thomas algorithm
%
% More explanation about this function and its syntax go here.
p=p_minus_2+2;
for k = 1:p-1
    i = k +1;
    l(i,k)=Ac(i,k)/Ac(k,k);
    for j = k :k+1
        Ac(i,j) = Ac(i,j)- l(i,k)*Ac(k,j);
    end
    b(i) = b(i)-l(i,k)*b(k);
end
%apply backward substitution
for k = p:-1:1
    g(k)=b(k);
    for j=k+1: min(p,k+1)
        g(k)=g(k)-Ac(k,j)*g(j);
    end
    g(k)=g(k)/Ac(k,k);
end
T_cn(2:n-1)=g(1:p);
end
    