rk2.m

来自「常微分方程的龙格库塔法系列程序」· M 代码 · 共 23 行

function rk2(f,u0,a,b,h)
%Runge_Kutta法,s=2
%调用格式:rk2(function_handle,u0,a,b,h)
%test: du/dt=u/2;u0=1;h=0.1;a=0;b=1;
%调用:先定义微分方程f=@(t,u) u/2;rk2(f,1,0,1,0.1)

%2007-03-15

tic;
len=1+(b-a)/h;
u=zeros(1,len);
t=a:h:b;
u(1)=u0;
for i=2:len,
u(i)=u(i-1)+h*(u(i-1)+h*f(t(i-1),u(i-1))/2)/2;
end
plot(t,u,'b+');
hold on;
line(t,u,'Color','b');
fun=@(t) exp(t/2);
fplot(fun,[0,1],'r');
error=sqrt(sum(abs(fun(t)-u).^2))
toc