Уравнения степени не выше второй — часть 3

def solve(*coefficients):
    roots = []
    lenght = len(coefficients)
    if lenght == 1:
        if coefficients[0] == 0:
            roots.append('all')
    if len(coefficients) == 2:
        b, c = coefficients[0], coefficients[1]
        roots.append(-c / b)
        print(*roots)
    if lenght == 3:
        a = coefficients[0]
        b = coefficients[1]
        c = coefficients[2]
        disc = b ** 2 - 4 * a * c
        if disc == 0:
            roots.append(-b / 2 * a)
        elif disc > 0:
            x1 = (-b + disc**0.5) / (a + a)
            x2 = (-b - disc**0.5) / (a + a)
            roots.append(x1)
            roots.append(x2)
    print(*roots)
 
 
lst = [int(i) for i in input().split()]
if len(lst) == 1:
    solve(lst[0])
if len(lst) == 2:
    solve(lst[0], lst[1])
if len(lst) == 3:
    solve(lst[0], lst[1], lst[2])

a = input().split(' ')
for t, i in enumerate(a):
    a[t] = int(i)
 
 
def solve(a):
    if len(a) == 3:
        d = a[1] ** 2 - 4 * a[0] * a[2]
        if a[0] == 0 and a[1] == 0 and a[2] == 0:
            x = ["all"]
        elif a[0] == 0 and a[1] == 0:
            x = ''
        elif a[0] == 0:
            x = [-a[2] / a[1]]
        elif a[1] == 0:
            x = [(a[2] // a[0]) ** 0.5]
        elif a[2] == 0:
            x = [0, -a[1] // a[0]]
        else:
            if d == 0:
                x = [-a[1] // (2 * a[0])]
            elif d < 0:
                x = ''
            else:
                x = [int((-a[1] + d ** 0.5) // (2 * a[0])),
                     int((-a[1] - d ** 0.5) // (2 * a[0]))]
        return x
    elif len(a) == 2:
        return [-a[1] / a[0]]
    elif len(a) == 1:
        if a[0] == 0:
            return ["all"]
        else:
            return []
    else:
        return ["all"]
 
 
print(*sorted(solve(a)))