x = [-b ± √(b² - 4ac)] / (2a)

import cmath  # 导入复数数学模块

def solve_quadratic(a, b, c):
    """
    求解二次方程 ax² + bx + c = 0
    返回方程的解
    """
    # 计算判别式
    discriminant = (b ** 2) - (4 * a * c)
    
    # 根据判别式的值计算根
    if discriminant > 0:
        # 两个不同的实数根
        root1 = (-b + cmath.sqrt(discriminant)) / (2 * a)
        root2 = (-b - cmath.sqrt(discriminant)) / (2 * a)
        return root1.real, root2.real, "两个不同的实数根"
    
    elif discriminant == 0:
        # 两个相同的实数根（重根）
        root = -b / (2 * a)
        return root, root, "两个相同的实数根"
    
    else:
        # 两个复数根
        root1 = (-b + cmath.sqrt(discriminant)) / (2 * a)
        root2 = (-b - cmath.sqrt(discriminant)) / (2 * a)
        return root1, root2, "两个复数根"

# 获取用户输入
print("二次方程式计算器: ax² + bx + c = 0")
a = float(input("请输入系数 a: "))
b = float(input("请输入系数 b: "))
c = float(input("请输入系数 c: "))

# 确保a不为零
if a == 0:
    print("错误：系数a不能为零！")
else:
    # 求解方程
    root1, root2, root_type = solve_quadratic(a, b, c)
    
    # 显示结果
    print("\n计算结果:")
    print(f"方程: {a}x² + {b}x + {c} = 0")
    print(f"根的类型: {root_type}")
    print(f"根1: {root1}")
    print(f"根2: {root2}")