function Z = heat(D,u0,q,tdim) xdim = length(u0); Z = zeros(tdim,xdim); Z(1,:) = u0; for tin = 2:tdim tip = tin - 1; Z(tin,2:end-1) = Z(tip,2:end-1) + ... D*q*(Z(tip,1:end-2)-2*Z(tip,2:end-1)+Z(tip,3:end)); end