คณิตศาสตร์เชิงตัวเลขเบื้องต้น
นี่คือ "สวัสดีโลก!" ของ PDEs (สมการเชิงอนุพันธ์ย่อย) Laplace หรือ Diffusion Equation ปรากฏบ่อยครั้งในวิชาฟิสิกส์เช่น Heat Equation, Deforming, Fluid Dynamics ฯลฯ ... เนื่องจากชีวิตจริงเป็น 3 มิติ แต่เราต้องการพูดว่า "Hello, World!" และไม่ร้องเพลง "99 ขวดเบียร์ ... " งานนี้มอบให้ใน 1D คุณอาจตีความว่าสิ่งนี้เป็นเสื้อคลุมยางผูกติดกับผนังทั้งสองด้านด้วยแรงบางอย่างที่ใช้กับมัน
ใน[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-based:{(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)
การใช้ทางเลือกโดยไม่มีเมทริกซ์พีชคณิต (ใช้วิธี Jacobi )
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)
ผมขอแนะนำให้เพิ่มตัวอย่างซึ่งไม่ได้แก้ปัญหาการวิเคราะห์เช่นกับ