mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-24 12:10:16 +00:00
38 lines
907 B
Python
38 lines
907 B
Python
from cmath import sqrt
|
|
from typing import Tuple
|
|
|
|
|
|
def quadratic_roots(a: int, b: int, c: int) -> Tuple[complex, complex]:
|
|
"""
|
|
Given the numerical coefficients a, b and c,
|
|
calculates the roots for any quadratic equation of the form ax^2 + bx + c
|
|
|
|
>>> quadratic_roots(a=1, b=3, c=-4)
|
|
(1.0, -4.0)
|
|
>>> quadratic_roots(5, 6, 1)
|
|
(-0.2, -1.0)
|
|
>>> quadratic_roots(1, -6, 25)
|
|
((3+4j), (3-4j))
|
|
"""
|
|
|
|
if a == 0:
|
|
raise ValueError("Coefficient 'a' must not be zero.")
|
|
delta = b * b - 4 * a * c
|
|
|
|
root_1 = (-b + sqrt(delta)) / (2 * a)
|
|
root_2 = (-b - sqrt(delta)) / (2 * a)
|
|
|
|
return (
|
|
root_1.real if not root_1.imag else root_1,
|
|
root_2.real if not root_2.imag else root_2,
|
|
)
|
|
|
|
|
|
def main():
|
|
solutions = quadratic_roots(a=5, b=6, c=1)
|
|
print("The solutions are: {} and {}".format(*solutions))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|