四阶阿当姆斯预测-校正公式.cpp
Upload User: hks123456
Upload Date: 2014-01-27
Package Size: 18k
Code Size: 1k
Category:

Algorithm

Development Platform:

Visual C++

  1. #include<iostream.h>
  2. float f(float x,float y)
  3. {
  4. return -x*y*y;
  5. }
  6. float *Runge_Kutta(float (*f)(float x,float y),float a,float b,float y0,int N)
  7. {
  8. float x=a,y=y0,K1,K2,K3,K4,*yy;
  9. float h=(b-a)/N;
  10. int i;
  11. yy=new float[(sizeof(float)*3)];
  12. for(i=1;i<=3;i++)
  13. {
  14. K1=f(x,y);
  15. K2=f(x+h/2,y+h*K1/2);
  16. K3=f(x+h/2,y+h*K2/2);
  17. K4=f(x+h,y+h*K3);
  18. y=y+h*(K1+2*K2+2*K3+K4)/6;
  19. x=a+i*h;
  20. *(yy+i-1)=y;
  21. }
  22. return yy;
  23. }
  24. void Adams(float a,float b,int N,float (*f)(float x,float y),float y0)
  25. {
  26. int i;
  27. float y1,y2,y,yp,yc,*yy,h,x;
  28. cout<<"x[0]="<<a<<'t'<<"    y[0]="<<y0<<endl;
  29. yy=Runge_Kutta(f,a,b,y0,N);
  30. y1=yy[0];
  31. y2=yy[1];
  32. y=yy[2];
  33. h=(b-a)/N;
  34. for(i=1;i<=3;i++)
  35. cout<<"x["<<i<<"]="<<a+i*h<<"    y["<<i<<"]="<<*(yy+i-1)<<endl;
  36. for(i=3;i<N;i++)
  37. {
  38. x=a+i*h;
  39. yp=y+h*(55*f(x,y)-59*f(x-h,y2)+37*f(x-2*h,y1)-9*f(x-3*h,y0))/24.0;
  40. yc=y+h*(9*f(x+h,yp)+19*f(x,y)-5*f(x-h,y2)+f(x-2*h,y1))/24.0;
  41. cout<<"x["<<i+1<<"]="<<x+h<<"    y["<<i+1<<"]="<<yc<<endl;
  42. y0=y1;
  43. y1=y2;
  44. y2=y;
  45. y=yc;
  46. }
  47. }
  48. void main()
  49. {
  50. float a=0,b=5.0,y0=2.0;
  51. int N=20;
  52. Adams(a,b,N,f,y0);
  53. }