求根是數值計算的一個基本問題,一般采用的都是迭代算法求解,主要有不動點迭代法、牛頓-拉富生算法、割線法和二分法。
- 不動點迭代法
所謂的不動點是指x=f(x)的那些點,而所謂的不懂點迭代法是指將原方程化為x=f(x)形式之后,下一步所用的x值為這一步的f(x),這樣的話就可以一直逼近我們需 要的x,即方程的根,但是這種方法可能不會收斂到方程的根,隨着初始值選定的大小,可能會有發散的情況,因此需要謹慎使用。
###不動點迭代法 func1 <- function(x){return(exp(exp(-x)))} fixpoint <- function(func, x0, tol=1e-8, max.iter=1e4){ ###求根的函數func ###初始值x0 ###允許誤差范圍tol ###最大循環次數max.iter x.old <- x0 x.new <- x0 for(i in 1:max.iter){ x.new <- func1(x.old) if(abs(x.new - x.old) < tol && i<max.iter){ cat('the iter time is',i,'\n') return(format(x.new,digits = 9)) } x.old <- x.new } cat('bad start num') }
- 牛頓-拉富生算法
所謂的牛頓-拉富生算法其實就是課本里面說的牛頓迭代法,也不是一個難的程序,主要思想就是x(n+1)=x(n)-f(x(n))/f`(x(n)),因此這種方法需要有求導表達式,限制了使用范圍。
###牛頓迭代法 ###示例方程 func1 <- function(x){return(log(x)-exp(-x))} func2 <- function(x){return(1/x+exp(-x))} Newton <- function(func1, func2, x0, tol=1e-8, max.iter=1e4){ ###牛頓迭代法// ###輸入方程式func1 ###輸入func1的導函數func2 ###初始值x0 ###誤差范圍tol ###最大迭代次數 x <- x0 for(i in 1:max.iter){ x <- x - func1(x)/func2(x) if(abs(x0-x) < tol){ cat('iter num is:',i,'\n') return(x) } x0 <- x } cat('this is a not good start num or the function is bad!') }
- 割線法
使用牛頓法的時候 ,需要先得到函數的導函數,但是很多時候都沒有導函數的解析表達式,因此可以考慮一種替代導函數的近似,即割線法,使用了一段直線作為函數某一段的近似,具體的數學表達式推導很簡單,在此略過,其主要假設是所求點的y值恰好為0,我是用相似比解出來的。
###割線法 ###割線法公式:x(n+1)=[f(xn)x(n-1) - x(n)f(x(n-1))] / [f(xn) - f(x(n-1))] ###示例函數 func1 <- function(x){return(log(x)-exp(-x))} gexian <- function(func,x0,x1,tol=1e-8,max.iter=1e4){ ###割線法需要兩個初始值 for(i in 1:max.iter){ x <- (func(x1)*x0 - x1*func(x0)) / (func(x1) - func(x0)) if(abs(x-x1) < tol){ cat('iter num is:',i,'\n') return(x) } x0 <- x1 x1 <- x } cat('this is a not good start num or the function is bad!') }
- 二分法
上述的方法均有可能不收斂,並且對函數會有一些額外的要求,一種比較暴力的方法但是適用范圍更廣的解決辦法就是二分法,其思想是如果f(x)連續,那么就一定會存在f(x1)<0,f(x2)>0的情況,根必然在x1和x2之間,之后只要不斷逼近根就好了。
###二分法 func1 <- function(x){return(log(x)-exp(-x))} half_M <- function(func,x0,x1,tol=1e-8,max.iter=1e4){ ###初始x0,x1必須滿足f(x0)f(x1) < 0,x0 < x1 for(i in 1:max.iter){ cat(x0," ",x1,"\n") if(abs(x0-x1) < tol){ cat('iter num:',i,"\n") return(x0) } temp_x <- (x0 + x1)/2 if((func(temp_x)*func(x0)) < 0){ x1 <- temp_x } else{ x0 <- temp_x } } }