欢迎关注更多精彩
关注我,学习常用算法与数据结构,一题多解,降维打击。
样条函数表示
根据力学原理,可以每两个固定点之间的函数二阶导数为一次函数,那么原来的函数就是分段(n+1个点有n段)三次函数。
解的条件
利用c2连续可以构建4n-2个方程,还需要人为添加2个条件才可以解出所有变量。
转化成三弯矩方程求解
上述方程可以使用追赶法求解。
有了M以后代入(1)式就得到了所有分段的函数。
追赶法解三对角矩阵
设原来三对角矩阵为A大小为n*n, 未知向量为M, 已经向量为V
追赶法算法如下:
void solve(A, M, V) {
for(int i=0;i<n-1;++i) {
b = A[i+1][i]/A[i][i];
A[i+1][i]=0;
A[i+1][i+1] -= A[i][i+1]*b;
V[i+1] -= V[i]*b;
}
// 经过上面的操作后,除了最后一行,每一行只有2个未知数。
// 再从最后一行开始依次求解出未知量
for(int i=n-1;i>=0;--i) {
M[i]=V[i]/A[i][i];
if(i==0)break;
V[i-1]-=M[i]*A[i-1][i];
A[i-1][i]=0;
}
}
效果演示
代码:
https://gitcode.com/chenbb1989/geometric_model/tree/master/gohw3
本人码农,希望通过自己的分享,让大家更容易学懂计算机知识。创作不易,帮忙点击公众号的链接,帮忙转发,感激不尽。