rk4.m

来自「余向阳博士Matlab教程的源代码。 包括：1.非线性方程的解法；2.线性方程」· M 代码 · 共 20 行

function R=rk4(f,a,b,ya,N)
%  y'=f(x,y)
%  a,b左右端点。
%  N为迭代步数。
%  h为步长。
%  ya为初值。
h=(b-a)/N;
T=zeros(1,N+1);
Y=zeros(1,N+1);
T=a:h:b;
Y(1)=ya;
for j=1:N
k1=h*feval(f,T(j),Y(j));
k2=h*feval(f,T(j)+h/2,Y(j)+k1/2);
k3=h*feval(f,T(j)+h/2,Y(j)+k2/2);
k4=h*feval(f,T(j)+h,Y(j)+k3);
Y(j+1)=Y(j)+(k1+2*k2+2*k3+k4)/6;
end
R=[T' Y']