python之初中数学题

初中数学题

题目描述：给你一个函数ax^2+bx+c=0

问你这个函数有多少个在实数域的解，并且把解给输出(因为解不一定是整数输出保留两位小数)

如果不存在合法解输出"wa"，如果有无穷多个解输出“inf”

输入

输入三个整数a,b,c

-100<=a<=100

-100<=c<=100

-100<=b<=100

输出

如果有解，先输出解得个数，然后后面输出解分别是什么?,有多个解得时候，解按照小到大输出

否则输出“wa“或者“inf”

from math import sqrt
a,b,c=map(int,input().split())
if a!=0:
    if b**2-4*a*c<0:
        print('wa')
    elif b**2-4*a*c==0:
        s=-b/(2*a)
        print('1 %.2f'%s)
    elif b**2-4*a*c>0:
        x=b**2-4*a*c
        s1=(-b+sqrt(x))/(2*a)
        s2=(-b-sqrt(x))/(2*a)
        if s1>s2:
            s1,s2=s2,s1
        print('2 %.2f %.2f'%(s1,s2))
    elif b==c==0:
        print('inf')
else:
    if b==c==0:
        print('inf')
    elif b==0 and c!=0:
        print('wa')
    elif b!=0:
        print('1 %.2f'%(-c/b))