Вступ до числової математики
Це "Привіт, світ!" PDE (часткових диференціальних рівнянь). Рівняння Лапласа або Дифузії часто з'являється у фізиці, наприклад, рівняння тепла, деформування, динаміка рідини тощо. Оскільки реальне життя є 3D, але ми хочемо сказати "Привіт, світ!" а не співати "99 пляшок пива, ..." це завдання дано в 1Д. Ви можете трактувати це як гумовий халат, прив'язаний до обох кінців до стіни з певною силою, прикладеної до нього.
У [0,1]
домені знайдіть функцію u
для заданої функції джерела f
та граничні значення, u_L
і u_R
таке:
-u'' = f
u(0) = u_L
u(1) = u_R
u''
позначає другу похідну від u
Це можна вирішити суто теоретично, але ваше завдання полягає в тому, щоб вирішити це чисельно на дискретному домені x для N
очок:
- на основі x =
{i/(N-1) | i=0..N-1}
або 1:{(i-1)/(N-1) | i=1..N}
h = 1/(N-1)
- це інтервал
Вхідні дані
f
як функція або вираз або рядокu_L
,u_R
як значення з плаваючою комоюN
як ціле число> = 2
Вихідні дані
- Array, List, якийсь відокремлений рядок
u
такого, щоu_i == u(x_i)
Приклади
Приклад 1
Вхідні дані : f = -2
, u_L = u_R = 0
, N = 10
(не брати f=-2
неправильно, це не значення , а функція константа , яка повертається -2
для всіх x
Це як постійна сила тяжіння на нашій мотузці.) .
Вихід: [-0.0, -0.09876543209876543, -0.1728395061728395, -0.22222222222222224, -0.24691358024691357, -0.24691358024691357, -0.22222222222222224, -0.1728395061728395, -0.09876543209876547, -0.0]
Існує просте точне рішення: u = -x*(1-x)
Приклад 2
Вхідні дані : f = 10*x
, u_L = 0
u_R = 1
, N = 15
(тут є багато навітряного на правій стороні)
Вихід: [ 0., 0.1898688, 0.37609329, 0.55502915, 0.72303207, 0.87645773, 1.01166181, 1.125, 1.21282799, 1.27150146, 1.29737609, 1.28680758, 1.2361516, 1.14176385, 1.]
Точне рішення для цього говорить: u = 1/3*(8*x-5*x^3)
Приклад 3
Вхідний сигнал: f = sin(2*pi*x)
, u_L = u_R = 1
, N = 20
(хто - то зламав тяжкості або є свого роду вгору у напрямку вітру)
Вихід: [ 1., 1.0083001, 1.01570075, 1.02139999, 1.0247802, 1.0254751, 1.02340937, 1.01880687, 1.01216636, 1.00420743, 0.99579257, 0.98783364, 0.98119313, 0.97659063, 0.9745249, 0.9752198, 0.97860001, 0.98429925, 0.9916999, 1.]
Тут точне рішення u = (sin(2*π*x))/(4*π^2)+1
Приклад 4
Вхід: f = exp(x^2)
, u_L = u_R = 0
,N=30
Вихід:
[ 0. 0.02021032 0.03923016 0.05705528 0.07367854 0.0890899
0.10327633 0.11622169 0.12790665 0.13830853 0.14740113 0.15515453
0.16153488 0.1665041 0.17001962 0.172034 0.17249459 0.17134303
0.16851482 0.1639387 0.15753606 0.1492202 0.13889553 0.12645668
0.11178744 0.09475961 0.07523169 0.05304738 0.02803389 0. ]
Зверніть увагу на незначну несиметрію
FDM
Один з можливих способів вирішити це - метод кінцевої різниці :
- переписати
-u_i'' = f_i
як (-u_{i-1} + 2u_i - u{i+1})/h² = f_i
що дорівнює-u_{i-1} + 2u_i - u{i+1} = h²f_i
- Встановіть рівняння:
- Які рівні рівняння матричного вектора:
- Розв’яжіть це рівняння і виведіть
u_i
Одна реалізація цього для демонстрації в Python:
import matplotlib.pyplot as plt
import numpy as np
def laplace(f, uL, uR, N):
h = 1./(N-1)
x = [i*h for i in range(N)]
A = np.zeros((N,N))
b = np.zeros((N,))
A[0,0] = 1
b[0] = uL
for i in range(1,N-1):
A[i,i-1] = -1
A[i,i] = 2
A[i,i+1] = -1
b[i] = h**2*f(x[i])
A[N-1,N-1] = 1
b[N-1] = uR
u = np.linalg.solve(A,b)
plt.plot(x,u,'*-')
plt.show()
return u
print laplace(lambda x:-2, 0, 0, 10)
print laplace(lambda x:10*x, 0, 1, 15)
print laplace(lambda x:np.sin(2*np.pi*x), 1, 1, 20)
Альтернативна реалізація без матричної алгебри (використовуючи метод Якобі )
def laplace(f, uL, uR, N):
h=1./(N-1)
b=[f(i*h)*h*h for i in range(N)]
b[0],b[-1]=uL,uR
u = [0]*N
def residual():
return np.sqrt(sum(r*r for r in[b[i] + u[i-1] - 2*u[i] + u[i+1] for i in range(1,N-1)]))
def jacobi():
return [uL] + [0.5*(b[i] + u[i-1] + u[i+1]) for i in range(1,N-1)] + [uR]
while residual() > 1e-6:
u = jacobi()
return u
Однак ви можете використовувати будь-який інший метод для розв’язання рівняння Лапласа. Якщо ви використовуєте ітеративний метод, слід повторити його до залишкового значення |b-Au|<1e-6
, b
будучи вектором правого бокуu_L,f_1h²,f_2h²,...
Примітки
Залежно від способу рішення, ви не можете вирішувати приклади саме до заданих рішень. Принаймні, щоб N->infinity
помилка повинна наближатися до нуля.
Стандартні лазівки заборонені , дозволено вбудовувати PDE.
Бонус
Бонус -30% за показ рішення, графічного або ASCII-art.
Перемога
Це codegolf, тому найкоротший код у байтах виграє!
log(log(x))
або sqrt(1-x^4)
які мають інтеграл, який, однак, не виражається в елементарних функціях.
u(x) = 1/2 (-sqrt(π) x erfi(x)+sqrt(π) erfi(1) x+e^(x^2)-e x+x-1)
він не є точно обчислюваним.
f(x) = exp(x^2)
.