mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-18 09:10:16 +00:00
9200a2e543
* from __future__ import annotations
* fixup! from __future__ import annotations
* fixup! from __future__ import annotations
* fixup! Format Python code with psf/black push
Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
39 lines
918 B
Python
39 lines
918 B
Python
from __future__ import annotations
|
|
|
|
from cmath import sqrt
|
|
|
|
|
|
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()
|