5.3习题

import cmath

class QuadraticEquation:
    def __init__(self,a ,b, c):
        self.a = a
        self.b = b
        self.c = c

    def getRoot(self):
        Delta = self.b ** 2 - 4 * self.a * self.c
        if Delta >= 0:
            root1 = (-self.b + Delta ** 0.5) / (2 * self.a)
            root2 = (-self.b - Delta ** 0.5) / (2 * self.a)
            return (root1, root2)
        else:
            root1 = (-self.b + cmath.sqrt(Delta)) / (2 * self.a)
            root2 = (-self.b - cmath.sqrt(Delta)) / (2 * self.a)
            return (root1, root2)

def main():
    a = float(input("请输入a的值: "))
    b = float(input("请输入b的值: "))
    c = float(input("请输入c的值: "))

    equation = QuadraticEquation(a , b , c)
    roots  = equation.getRoot()
    print("方程的根为: ")
    print(f"Root 1: {roots[0]}")
    print(f"Root 2: {roots[1]}")

if __name__ == "__main__":
    main()