二阶龙格库塔公式推导_数值常微分方程-欧拉法与龙格-库塔法

大三时候在跳蚤市场闲逛,从一位数学院的学长那里买了一些闲书,最近翻出来刚好有李荣华、刘播老师的《微分方程数值解法》和王仁宏老师的《数值逼近》,结合周善贵老师的《计算物理》课程,整理一下笔记。

本文整理常微分方程数值求解的欧拉法与龙格-库塔法。

一般地,动力学系统的时间演化可以用常微分方程的初值问题来描述,例如设一维简谐运动的回复力:

,有则运动方程:

。令

,可以将二阶微分方程转化为一阶微分方程组:

因此本文主要整理一阶常微分方程初值问题的数值解法。

一阶常微分方程初值问题

在区域

上连续,对于一个给定的常微分方程

及初值

,求解

。为了保证解

存在、唯一且连续依赖初值

,要求

满足Lipschitz条件:

存在常数L,使得

对所有

成立。

假设

总满足上述条件。常用的近似解法有级数解法等近似解析方法,以及下文整理的数值方法:欧拉法与龙格-库塔法。

欧拉法

将区间

作N等分,每一小区间长度

称为步长,

称为节点。根据初值

,代入微分方程可直接解出

的导数值

推导

1、根据泰勒展开式:

略去二阶小量,得:

以此类推,得到递推公式:

2、数值积分推导

可得:

,使用左矩形积分得:

以此类推,可得到:

为了提高精度,可以使用梯形积分代替矩形积分,即:

以此类推,得到改进的欧拉法:

Python计算实例

为例,其精确解为

,使用欧拉法求解的Python代码如下:

import math

from matplotlib import pyplot as plt

t_0 = 0

y_0 = 1

tau = 0.1

i = 1

solve = []

Euler = []

t = []

while i < 100:

if i == 1:

y_n = y_0

t_n = t_0

Euler.append(y_n)

solve.append(math.exp(t_n))

t.append(t_n)

func = y_n

y_n = y_n + tau * func

t_n = t_n + tau

i += 1

plt.plot(t, Euler, c='green', label=' Euler method')

plt.plot(t, solve, c='red', label=' accuracy')

plt.fill_between(t, solve, Euler, facecolor='blue', alpha=0.2)

plt.title('Euler method', fontsize=19)

plt.xlabel('t', fontsize=19)

plt.ylabel('y', fontsize=19)

plt.legend()

plt.show()

作图可以看到,当迭代步数较多后,欧拉法的结果逐渐落后于精确指数解的增长速度。下面分析欧拉法的误差来源。

中,略去了高阶小量

,因此在每一步的递推中,都有局部截断误差

,其阶为

在计算中,我们更关心精确解和数值解之间的误差

,称为整体误差,其满足

根据Lipschitz条件,可得:

,可得:

局部截断误差

的二阶量,设

,得:

,整体误差是

的一阶量。同理可得,改进的欧拉法局部截断误差

的三阶量

,整体误差是

的二阶量

稳定性分析

如果计算的初值不能精确给定,例如存在测量、舍入误差等,在计算过程中,每一步传递的误差连续依赖于初始误差,则称算法稳定,否则该算法不稳定。

对于不同的初值

,有

两式相减,得:

根据Lipschitz条件,可得:

连续依赖于初始误差,欧拉法稳定。同理,改进的欧拉法也稳定。

龙格-库塔法

龙格库塔法的主要思想:在

点的附近选取一些特定的点,然后把这些点的函数值进行线性组合,使用组合值代替泰勒展开中

点的导数值。

泰勒展开:

,根据多元函数求导法则有:

以此类推,可以得到:

同时,我们可以写出泰勒展开的形式解:

其中:

通项为:

基本思路是,利用当前点的函数值

可以计算出

,然后引入参数

和步长

可以计算出

,之后使用

和步长

计算

,以此类推,直到

现在把

展开:

代入得:

代入

可得:

与泰勒展开式

相比较,可知:

2个方程有3个未知数,因此有无穷多个解,可采用

,则:

,可以改写为:

此即为二阶龙格-库塔法。

与上一节的欧拉法公式对比:

,因此二阶龙格-库塔法取参数

时,即为改进的欧拉法。

Python计算实例

仍以

为例,其精确解为

,使用二阶龙格-库塔法求解的Python代码如下:

import math

from matplotlib import pyplot as plt

t_0 = 0

y_0 = 1

z_0 = 1

tau = 0.1

i = 1

j = 1

solve = []

Euler = []

R_K = []

t = []

while i < 100:

if i == 1:

y_n = y_0

t_n = t_0

R_K.append(y_n)

solve.append(math.exp(t_n))

t.append(t_n)

func_n = y_n

func_m = y_n + tau * func_n

y_n = y_n + 0.5 * tau * (func_n + func_m)

t_n = t_n + tau

i += 1

t = []

while j < 100:

if j == 1:

z_n = z_0

t_n = t_0

Euler.append(z_n)

t.append(t_n)

func = z_n

z_n = z_n + tau * func

t_n = t_n + tau

j += 1

plt.scatter(t, R_K, marker='^', c='blue', s=70, label=' R-K method')

plt.plot(t, Euler, c='green', label=' Euler method')

plt.plot(t, solve, c='red', label=' accuracy')

plt.fill_between(t, solve, Euler, facecolor='yellow', alpha=0.2)

plt.title('Euler method & R-K method', fontsize=19)

plt.xlabel('t', fontsize=19)

plt.ylabel('y', fontsize=19)

plt.legend()

plt.show()

黄色部分表示数值解和精确解的偏离,可以看到,二阶龙格-库塔法(改进的欧拉法)精确度得到了很大的提升。

二阶龙格-库塔法中,泰勒展开到了

阶,通过与泰勒展开系数进行对比,可以得到含3个未知数的2个方程。依次类推,如果泰勒展开到了

阶,对比

可以得到

阶龙格-库塔法。常用经典四阶龙格-库塔法:

Reference:

1、周善贵,《计算物理》课程讲义

2、李荣华,刘播,《微分方程数值解法》

3、王仁宏,《数值逼近》

你可能感兴趣的:(二阶龙格库塔公式推导)