heun.m

来自「国外计算机科学教材系列 numerial method using matla」· M 代码 · 共 32 行

function H=heun(f,a,b,ya,M)

%Input   - f is the function entered as a string 'f'
%        - a and b are the left and right endpoints
%        - ya is the initial condition y(a)
%        - M is the number of steps
%Output  - H=[T' Y'] where T is the vector of abscissas and
%          Y is the vector of ordinates

% NUMERICAL METHODS: MATLAB Programs
%(c) 1999 by John H. Mathews and Kurtis D. Fink
%To accompany the textbook:
%NUMERICAL METHODS Using MATLAB,
%by John H. Mathews and Kurtis D. Fink
%ISBN 0-13-270042-5, (c) 1999
%PRENTICE HALL, INC.
%Upper Saddle River, NJ 07458

h=(b-a)/M;
T=zeros(1,M+1);
Y=zeros(1,M+1);
T=a:h:b;
Y(1)=ya;
for j=1:M
   k1=feval(f,T(j),Y(j));
   k2=feval(f,T(j+1),Y(j)+h*k1); 
   Y(j+1)=Y(j)+(h/2)*(k1+k2);
end

H=[T' Y'];

% ss=heun('f',0,3,1,20),plot(ss(:,1),ss(:,2))