mirror of
https://hub.njuu.cf/TheAlgorithms/Python.git
synced 2023-10-11 13:06:12 +08:00
Implemented Square Root Algorithm (#1687)
* Added to maths and strings * added changes suggest by cclauss * added square root function * Fixed type hinting * fixed type error * Fixed another type error
This commit is contained in:
parent
fc4c0f5bfd
commit
38bad6b1e8
63
maths/square_root.py
Normal file
63
maths/square_root.py
Normal file
@ -0,0 +1,63 @@
|
||||
import math
|
||||
|
||||
|
||||
def fx(x: float, a: float) -> float:
|
||||
return math.pow(x, 2) - a
|
||||
|
||||
|
||||
def fx_derivative(x: float) -> float:
|
||||
return 2 * x
|
||||
|
||||
|
||||
def get_initial_point(a: float) -> float:
|
||||
start = 2.0
|
||||
|
||||
while start <= a:
|
||||
start = math.pow(start, 2)
|
||||
|
||||
return start
|
||||
|
||||
|
||||
def square_root_iterative(
|
||||
a: float, max_iter: int = 9999, tolerance: float = 0.00000000000001
|
||||
) -> float:
|
||||
"""
|
||||
Sqaure root is aproximated using Newtons method.
|
||||
https://en.wikipedia.org/wiki/Newton%27s_method
|
||||
|
||||
>>> all(abs(square_root_iterative(i)-math.sqrt(i)) <= .00000000000001 for i in range(0, 500))
|
||||
True
|
||||
|
||||
>>> square_root_iterative(-1)
|
||||
Traceback (most recent call last):
|
||||
...
|
||||
ValueError: math domain error
|
||||
|
||||
>>> square_root_iterative(4)
|
||||
2.0
|
||||
|
||||
>>> square_root_iterative(3.2)
|
||||
1.788854381999832
|
||||
|
||||
>>> square_root_iterative(140)
|
||||
11.832159566199232
|
||||
"""
|
||||
|
||||
if a < 0:
|
||||
raise ValueError("math domain error")
|
||||
|
||||
value = get_initial_point(a)
|
||||
|
||||
for i in range(max_iter):
|
||||
prev_value = value
|
||||
value = value - fx(value, a) / fx_derivative(value)
|
||||
if abs(prev_value - value) < tolerance:
|
||||
return value
|
||||
|
||||
return value
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
from doctest import testmod
|
||||
|
||||
testmod()
|
Loading…
Reference in New Issue
Block a user