print("Program na výpočet kořenů kvadratické rovnice a*x^2 + bx + c. \nProgram ukončíte stiskem klávesy 'enter'.\n")
print("Tvar rovnice: ax^2+bx+c=0")
a = float(input("Zadej koencifient a: "))
b = float(input("Zadej koencifient b: "))
c = float(input("Zadej koencifient c: "))
if a != 0: #ošetření, zda a není nula (pak to není kvadratická rovnice)
    d = b*b - 4*a*c #vypočteme si diskriminant
    if d < 0:
        x1 = (-b + d**0.5)/(2*a) #využijeme, že python umí pracovat s komplexními čísly, a tedy s odmocninou ze záporného čísla
        x2 = (-b - d**0.5)/(2*a)
        print("Rovnice má dva komplexní kořeny: ")
        print(f'{x1 = :.2f}') # vylepšené print(x1): formátovaný výstup: vypíše x1 na dvě desetinná místa
        print(f'{x2 = :.2f}')
    elif d == 0:
        x = -b/(2*a)
        print("Rovnice má jeden kořen: ")
        print(f'{x = :.2f}')
    else:
        x1 = (-b + d**0.5)/(2*a)
        x2 = (-b - d**0.5)/(2*a)
        print("Rovnice má dva reálné kořeny: ")
        print(f'{x1 = :.2f}')
        print(f'{x2 = :.2f}')
else:
    print("Není kvadratická rovnice")
input()