백준 9735번: 삼차 방정식 풀기 (Python)

9735번: 삼차 방정식 풀기

다항식을 더 쉽게 풀고 싶다면? 

조립제법 - 위키백과, 우리 모두의 백과사전

???: 근은 구했는데 어떻게 품?

import sys
sys.stdin.readline

n = int(input())
for i in range(n):
    A, B, C, D = map(int, input().split())
    root = [] # 근
    k = 0 # 정수해
    for num in range(-1000000, 1000001):
        if A*(num**3) + B*(num**2) + C*num + D == 0:
            k = num
            root.append(k)
            break
    a = A
    b = B+A*k
    c = C+B*k+A*(k**2)
    dis = b**2 - 4*a*c
    if dis == 0:
        root.append(-b/(2*a))
    elif dis > 0:
        root.append((-b+(b**2-4*a*c)**(0.5)) / (2*a))
        root.append((-b-(b**2-4*a*c)**(0.5)) / (2*a))
    else:
        pass
    root.sort()
    root = list(set(root))
    for i in range(len(root)):
        print(f"{round(root[i],4)}", end=' ')
    print()

시간제한이 2초니까 정수해는 for문으로 찾고 실수해는 근의 공식을 이용해서 찾으면 된다.