编写程序，解一元二次方程 ax2+bx+c=0

from math import *

a = float(input("a:"))

b = float(input("b:"))

c = float(input("c:"))

if a == 0:

print("x=",-c/b)

else:

beta = b**2-4*a*c #b**2是b的平方

if beta > 0:

x1 = (-b + sqrt(beta))/(2*a) #记得大括号

x2 = (-b - sqrt(beta))/(2*a)

print("x1=",x1," x2=",x2)

elif beta == 0:

x = (-b + sqrt(beta))/(2*a)

print("x=",x)

else:

real = -b/(2*a)

imag = beta/(2*a)

print("x1=",complex(real,imag),"x2=",complex(real,-imag))