Files
sjy01-image-proc/vendor/github.com/nuknal/goNum/Secant2P.go
2024-10-24 15:46:01 +08:00

80 lines
2.0 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

// Secant2P
/*
------------------------------------------------------
作者 : Black Ghost
日期 : 2018-11-02
版本 : 0.0.0
------------------------------------------------------
双点弦截法求解方程 f(x)=0 在区间[a, b]内的根
理论:
1. 当xE[a, b]f''(x)连续f'(x) != 0
xn0*f(xn1) - xn1*f(xn0)
xn2 = -------------------------
f(xn1) - f(xn0)
超线性收敛,收敛阶(1+5^0.5)/2
------------------------------------------------------
输入 :
fn f(x)函数定义为等式左侧部分右侧为0
a, b 求解区间
N 步数上限
tol 误差上限
输出 :
sol 解值
err 解出标志false-未解出或达到步数上限;
true-全部解出
------------------------------------------------------
*/
package goNum
import (
"math"
)
// Secant2P 双点弦截法求解方程 f(x)=0 在区间[a, b]内的根
func Secant2P(fn func(float64) float64, a, b float64,
N int, tol float64) (float64, bool) {
/*
双点弦截法求解方程 f(x)=0 在区间[a, b]内的根
输入 :
fn f(x)函数定义为等式左侧部分右侧为0
a, b 求解区间
N 步数上限
tol 误差上限
输出 :
sol 解值
err 解出标志false-未解出或达到步数上限;
true-全部解出
*/
var sol float64
var err bool = false
//判断a b的次序
if (b < a) || (fn(a)*fn(b) > 0) {
return sol, err
}
// 求解
sol = (a*fn(b) - b*fn(a)) / (fn(b) - fn(a))
for i := 0; i < N; i++ {
//判断是否解得
if (fn(a)*fn(sol) > 0) && (math.Abs(sol-a) < tol) {
err = true
return sol, err
} else if (fn(a)*fn(sol) < 0) && (math.Abs(sol-b) < tol) {
err = true
return sol, err
}
//下一步
switch {
case fn(a)*fn(sol) > 0:
a = sol
default:
b = sol
}
sol = (a*fn(b) - b*fn(a)) / (fn(b) - fn(a))
}
return sol, err
}