euler_t.m

来自「一种数值分析常用的欧拉算法,用matlab实现的」· M 代码 · 共 25 行

function [x,y]=euler_t(ydot_fun, x0, y0, h, N)
% 梯形公式，其中
%    ydot_fun ---  一阶微分方程的函数
%    x0, y0   ---  初始条件
%    h        ---  区间步长
%    N        ---  区间的个数
%    x        ---  Xn 构成的向量
%    y        ---  Yn 构成的向量

x=zeros(1,N+1); y=zeros(1,N+1); x(1)=x0; y(1)=y0;
for n=1:N
    x(n+1)=x(n)+h;
    % 用迭代法求y(n+1)
    z0=y(n)+h*feval(ydot_fun, x(n), y(n));
    for k=1:3
        z1=y(n)+h/2*(feval(ydot_fun, x(n), y(n)) ... 
               +feval(ydot_fun, x(n+1), z0));
        if abs(z1-z0)<1e-3
            break;
        end
        z0=z1;
    end
    y(n+1)=z1;
end