Advertisement
EconomicSerg

Bisect

Oct 22nd, 2020
297
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Lua 0.67 KB | None | 0 0
  1. -- bisect.lua
  2. -- bisection method for solving non-linear equations
  3.  
  4. delta=1e-6  -- tolerance
  5.  
  6. function bisect(f,a,b,fa,fb)
  7.  local c=(a+b)/2
  8.  io.write(n," c=",c," a=",a," b=",b,"\n")
  9.  if c==a or c==b or math.abs(a-b)<delta then return c,b-a end
  10.  n=n+1
  11.  local fc=f(c)
  12.  if fa*fc<0 then return bisect(f,a,c,fa,fc) else return bisect(f,c,b,fc,fb) end
  13. end
  14.  
  15. -- find root of f in the inverval [a,b]. needs f(a)*f(b)<0
  16. function solve(f,a,b)
  17.  n=0
  18.  local z,e=bisect(f,a,b,f(a),f(b))
  19.  io.write(string.format("after %d steps, root is %.17g with error %.1e, f=%.1e\n",n,z,e,f(z)))
  20. end
  21.  
  22. -- our function
  23. function f(x)
  24.  return x*x*x-x-1
  25. end
  26.  
  27. -- find zero in [1,2]
  28. solve(f,1,2)
  29.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement