一元二次方程求解

import math
import cmath

print("请输入一元二次方程的三个系数ABC：")
A = float(input("A = "))
B = float(input("B = "))
C = float(input("C = "))
if A == 0:
    if B == 0:
        if C != 0:
            print("无解！")
        else:
            print("x为无穷多组任意解")
    else:
        print("x = ", -C / B)
else:
    die = B * B - 4 * A * C
    if die < 0:
        print("在实数范围内无解")
        x1 = complex(-B / (2 * A), complex(0,cmath.sqrt(die) / (2 * A)))
        x2 = complex(-B / (2 * A), complex(0,-cmath.sqrt(die) / (2 * A)))
        print("在复数范围内的解如下")
        print("x1 =", x1)
        print("x2 =", x2)
    elif die == 0:
        print("x1 = x2 =", -B / (2 * A))
    else:
        x1 = (-B + math.sqrt(die)) / (2 * A)
        x2 = (-B - math.sqrt(die)) / (2 * A)
        print("x1 =", x1)
        print("x2 =", x2)