TDT4109/Exercise 2/11 - Andregradsligning/task11b.py

from math import sqrt

from task11a import getValues


if __name__ == "__main__":
  a, b, c = getValues()
  d = b**2 - 4 * a * c

  expression = f'{a}x^2 + {b}x + {c}'
  if d > 0:
    roots = (
      (-b + sqrt(d)) / (2 * a),
      (-b - sqrt(d)) / (2 * a)
    )
    print(
        f'Andregradsligningen {expression} har de to reelle løsningene {roots[0]} og {roots[1]}'
    )
  elif d == 0:
    root = (-b + sqrt(d)) / (2 * a)
    print(f'Andregradsligningen {expression} har en reell dobbelrot {root}')
  else:
    print(f'Andregradsligningen {expression} har to imaginære løsninger')