rk4.m

来自「这是用来解常微分方程的原代码」· 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'];