solve_tridiag.m

function matX = solve_tridiag(ud,uo,lo,b)

%Define matrices
matX = zeros(length(b),1);
matY = zeros(length(b),1);

matY(1) = b(1);

for i = 2:length(b)
    matY(i)= b(i) - lo(i-1)*matY(i-1);
end

matX(length(b)) = matY(length(b))/ud(length(b));

for i = length(ud)-1:-1:1
    matX(i) = (matY(i)-uo(i)*matX(i+1))/ud(i);

end
end