引言
考虑存在以下二阶偏微分方程
\[\begin{align} f_2 \cdot \ddot{X(t)}+f_1 \cdot \dot{X(t)} +f_0 \cdot {X(t)} =F(t) \end{align} \]
如何使用四阶龙格-库塔法求解该微分方程？
一阶微分方程的解法
首先回顾下对于一阶微分方程的解法，现在有以下一阶常系数非齐次微分方程
\[\begin{align} f_1 \cdot \dot{X(t)} +f_0 \cdot {X(t)} =F(t) \end{align} \]
方程可以写作
\[\begin{align} \dot{X(t)} = \frac {F(t)-f_0 \cdot {X(t)}}{f_1 }\\ \frac {X_{n+1}-X_n}{dt} = \frac {F(t)-f_0 \cdot {X(t)}}{f_1 }\\ X_{n+1}=X_n+dt\cdot\frac {F(t)-f_0 \cdot {X_{n}}}{f_1 }=X_n+dt\cdot f(t,X_n) \end{align} \]
其中\(dt\)为时间步长。
四阶龙格-库塔法公式如下：
\[\begin{align} \begin{cases} X_{n+1}=X_{n}+\frac{dt}{6}\left (k_1+2k_2+2k_3+k_4 \right) \\ k_1=f\left (t,X_n \right) \\ k_2=f\left (t+\frac {dt}{2},X_n+\frac {dt}{2}\cdot k_1 \right) \\ k_3=f\left (t+\frac {dt}{2},X_n+\frac {dt}{2}\cdot k_2 \right) \\ k_4=f\left (t+{dt},X_n+dt \cdot {k_3} \right)\\ \end{cases} \end{align} \]
利用以上公式即可显式推进求解
二阶微分方程的解法
现在考虑二阶常系数非齐次微分方程
\[\begin{align} f_2 \cdot \ddot{X(t)}+f_1 \cdot \dot{X(t)} +f_0 \cdot {X(t)} =F(t) \end{align} \]
方程中出现二次导数项\(\ddot{X(t)}\)，难以处理，所以考虑将二次导数项\(\ddot{X(t)}\)转化为一次导数项，至此，我们引入
\[\begin{align} \begin{cases} a(t)=X(t) \\ b(t)= \dot a(t)=\dot X(t) \end{cases} \end{align} \]
原方程可以写成一阶偏微分方程组
\[\begin{align} \begin{cases} f_2\cdot \dot b(t)+f_1\cdot b(t) +f_0\cdot a(t)= F(t) \\ b(t)=\dot a(t) \end{cases} \end{align} \]
仿照以上一阶微分方程的RK4解法
\[\begin{align} \begin{cases} \dot a(t)=b(t)\\ \dot b(t)= \frac {F(t)-(f_1\cdot b(t) +f_0\cdot a(t))}{f_2} \\ \end{cases} \end{align} \]
\[\begin{align} \begin{cases} \dot a_{n+1}=a_{n}+dt\cdot b(t) = a_{n}+dt\cdot f(t,a_n,b_n) \\ \dot b_{n+1}=b_{n}+dt\cdot \frac {F(t)-(f_1\cdot b(t) +f_0\cdot a(t))}{f_2} = b_{n}+dt\cdot g(t,a_n,b_n) \\ \end{cases} \end{align} \]
\[\begin{align} \begin{cases} \dot a_{n+1}=a_{n}+dt\cdot f(t,a_n,b_n) \\ \dot b_{n+1} = b_{n}+dt\cdot g(t,a_n,b_n) \\ \end{cases} \end{align} \]
化简至此，便可以仿照之前的RK4方法进行求解，具体公式为
\[\begin{bmatrix} a_{n+1}\\b_{n+1} \end{bmatrix}= \begin{bmatrix} a_n+\frac{dt}{6}\cdot (k_{1a}+2k_{2a}+2k_{3a}+k_{4a})\\ b_n+\frac{dt}{6}\cdot (k_{1b}+2k_{2b}+2k_{3b}+k_{4b}) \end{bmatrix}\\ \]
其中，各个系数求解公式为：
\[\begin{align} \begin{cases} \begin{bmatrix} k_{1a}\\k_{1b} \end{bmatrix}= \begin{bmatrix} f(t,a_n,b_n)\\ g(t,a_n,b_n) \end{bmatrix}\\ \begin{bmatrix} k_{2a}\\k_{2b} \end{bmatrix}= \begin{bmatrix} f(t+\frac{dt}{2},a_n+\frac{dt}{2} \cdot k_{1a},b_n+\frac{dt}{2} \cdot k_{1b})\\ g(t+\frac{dt}{2},a_n+\frac{dt}{2} \cdot k_{1a},b_n+\frac{dt}{2} \cdot k_{1b}) \end{bmatrix}\\ \begin{bmatrix} k_{3a}\\k_{3b} \end{bmatrix}= \begin{bmatrix} f(t+\frac{dt}{2},a_n+\frac{dt}{2} \cdot k_{2a},b_n+\frac{dt}{2} \cdot k_{2b})\\ g(t+\frac{dt}{2},a_n+\frac{dt}{2} \cdot k_{2a},b_n+\frac{dt}{2} \cdot k_{2b}) \end{bmatrix}\\ \begin{bmatrix} k_{4a}\\k_{4b} \end{bmatrix}= \begin{bmatrix} f(t+dt,a_n+dt \cdot k_{3a},b_n+dt \cdot k_{3b})\\ g(t+dt,a_n+dt \cdot k_{3a},b_n+dt \cdot k_{3b}) \end{bmatrix}\\ \end{cases} \end{align} \]
至此已经完成使用四阶龙格-库塔法二阶微分方程求解过程的梳理
实例
求解，以下偏微分方程：
\[\begin{align} 1 \cdot \ddot{X(t)}+0 \cdot \dot{X(t)} +0 \cdot {X(t)} =cos(t)\\ \dot{X(0)}=0\\ {X(0)} =0 \end{align} \]
理论解：
\[\begin{align} {X(t)} =-cos(t)+1 \end{align} \]
#include"RK_ODE.h"
#include<iostream>
using namespace std;
#define M 1
MatrixXd F2(double t) {
MatrixXd res = MatrixXd::Identity(M, M);
return res;
}
MatrixXd F1(double t) {
MatrixXd res = MatrixXd::Zero(M, M);
return res;
}
MatrixXd F0(double t) {
MatrixXd res = MatrixXd::Zero(M, M);
return res;
}
MatrixXd F(double t) {
MatrixXd res = MatrixXd::Ones(M, 1);
res(0, 0) = cos(t);
return res;
}
int main() {
RK_ODE ode(M, 0.1);
ode.X.fill(0.);
ode.dX.fill(0.);
for (int i = 0; i < 1000; i++) {
ode.Solve(F2, F1, F0, F);
ode.WriteMotionAndForce("results1.txt", 0);
}
//ode.SaveSnapshoot("snapshoot.json");
cout << ode._Mode << endl;
}