Computer >> Máy Tính >  >> Lập trình >> Lập trình

Quy tắc bậc 4 Runge-Kutta cho phương trình vi phân


Phương pháp Runge Kutta được sử dụng để giải phương trình vi phân thông thường (ODE). Nó sử dụng hàm dy / dx cho x và y, đồng thời cũng cần giá trị ban đầu của y, tức là y (0). Nó tìm giá trị gần đúng của y cho x đã cho. Để giải quyết ODE, chúng ta phải làm theo các công thức sau:

Quy tắc bậc 4 Runge-Kutta cho phương trình vi phân

Đây là chiều cao của khoảng thời gian.

Lưu ý: Từ các công thức này, chúng ta có thể sử dụng hai k1 và k2 đầu tiên để tìm giải pháp bậc 2 Runge-Kutta cho ODE.

Đầu vào và Đầu ra

Input:
The x0 and f(x0): 0 and 0
the value of x = 0.4
the value of h = 0.1
Output:
Answer of differential equation: 0.0213594

Thuật toán

rungeKutta(x0, y0, x, h)

Đầu vào - Giá trị x và y ban đầu, giá trị x được nhắm mục tiêu và chiều cao của khoảng h.

Đầu ra - Giá trị của y cho giá trị x.

Begin
   iteration := (x – x0)/h
   y = y0
   for i := 1 to iteration, do
      k1 := h*f(x0, y)
      k2 := h*f((x0 + h/2), (y + k1/2))
      k3 := h*f((x0 + h/2), (y + k2/2))
      k4 := h*f((x0 + h), (y + k3))
      y := y + (1/6)*(k1 + 2k2 + 2k3 + k4)
      x0 := x0 + h
   done
   return y
End

Ví dụ

#include <iostream>
using namespace std;

double diffOfy(double x, double y) {
   return ((x*x)+(y*y)); //function x^2 + y^2
}

double rk4thOrder(double x0, double y0, double x, double h) {
   int iteration = int((x - x0)/h);    //calculate number of iterations
   double k1, k2, k3, k4;
   double y = y0;    //initially y is f(x0)

   for(int i = 1; i<=iteration; i++) {
      k1 = h*diffOfy(x0, y);
      k2 = h*diffOfy((x0+h/2), (y+k1/2));
      k3 = h*diffOfy((x0+h/2), (y+k2/2));
      k4 = h*diffOfy((x0+h), (y+k3));
         
      y += double((1.0/6.0)*(k1+2*k2+2*k3+k4));    //update y using del y
      x0 += h;    //update x0 by h
   }
   return y;    //f(x) value
}

int main() {
   double x0, y0, x, h;
   cout << "Enter x0 and f(x0): "; cin >> x0 >> y0;
   cout << "Enter x: "; cin >> x;
   cout << "Enter h: "; cin >> h;
   cout << "Answer of differential equation: " << rk4thOrder(x0, y0, x, h);
}

Đầu ra

Enter x0 and f(x0): 0 0
Enter x: 0.4
Enter h: 0.1
Answer of differential equation: 0.0213594