【编程入门题--自定义函数求一元二次方程】

题目描述

求方程 的根，用三个函数分别求当b^2-4ac大于0、等于0、和小于0时的根，并输出结果。从主函数输入a、b、c的值。

示例

输入格式

a b c

输出格式

x1=? x2=?

样例输入

4 1 1

样例输出

x1=-0.125+0.484i x2=-0.125-0.484i

代码

from math import sqrt

def f1(a,b,c):
    x1=((-b)+sqrt(v))/(2*a)
    x2=((-b)-sqrt(v))/(2*a)
    return x1,x2

def f2(a,b,c):
    x = (-b) / (2 * a)
    x1=x
    x2=x
    return x1,x2

def f3(a,b,c):
    x = (-b) / (2 * a)
    m = sqrt(-v) / (2 * a)
    n = -sqrt(-v) / (2 * a)
    m="{:.3f}".format(m)
    n = "{:.3f}".format(n)
    x1=str(x)+"+"+str(m)+'i'
    x2=str(x)+str(n)+'i'
    return x1, x2

if __name__ == '__main__':
    a,b,c=map(int,input().split())
    v=b*b-4*a*c
    if v>0:
        x1,x2=f1(a,b,v)
    elif v==0:
        x1,x2=f2(a,b,v)
    else:
        x1,x2=f3(a,b,v)
    print(f'x1={x1} x2={x2}')