记录PTA编程练习题7-18 二分法求多项式单根——Python3实现

题目描述：

        二分法求函数根的原理为：如果连续函数f(x)在区间[a,b]的两个端点取值异号，即f(a)f(b)<0，则它在这个区间内至少存在1个根r，即f(r)=0。

二分法的步骤为：

        1. 检查区间长度，如果小于给定阈值，则停止，输出区间中点(a+b)/2；否则

        2. 如果f(a)f(b)<0，则计算中点的值f((a+b)/2)；

        3. 如果f((a+b)/2)正好为0，则(a+b)/2就是要求的根；否则

        4. 如果f((a+b)/2)与f(a)同号，则说明根在区间[(a+b)/2,b]，令a=(a+b)/2，重复循环；

        5. 如果f((a+b)/2)与f(b)同号，则说明根在区间[a,(a+b)/2]，令b=(a+b)/2，重复循环。

本题目要求编写程序，计算给定3阶多项式f(x)=a3​x3+a2​x2+a1​x+a0​在给定区间[a,b]内的根。

输入格式：

        输入在第1行中顺序给出多项式的4个系数a3​、a2​、a1​、a0​，在第2行中顺序给出区间端点a和b。题目保证多项式在给定区间内存在唯一单根。

输出格式：

        在一行中输出该多项式在该区间内的根，精确到小数点后2位。

代码实现：

        根据上面的5个条件编写代码。

a3,a2,a1,a0 = map(float,input().split())
a,b = map(float,input().split())
def f(x):  #f(x)函数
    return a3*x**3 + a2*x**2 + a1*x + a0
def q(a,b):  #二分法求根函数
    if b-a < 0.01:  #条件1，设定阈值，这里设为0.01
        print("{:.2f}".format((a+b)/2))
    elif f(a) * f(b) < 0:  #条件2
        if f((a+b)/2) == 0:  #条件3
            print("{:.2f}".format((a+b)/2))
        else:
            if f((a+b)/2)*f(a) > 0:  #条件4
                a = (a+b)/2
                q(a,b)
            elif f((a+b)/2)*f(b) > 0:  #条件5
                b = (a+b)/2
                q(a,b)
    elif f(a) == 0:
        print("{:.2f}".format(a))
    elif f(b) == 0:
        print("{:.2f}".format(b))
if a<b:  #如果满足a<b,则执行二分法求根函数
    q(a,b)
elif a==b:
    print("{:.2f}".format(a))