글 읽기 - 벌집문제 시간초과,,, 도저히 모르겠네요

ie1914 4년 전 0

3시간동안 질문 다 찾아보고 온갖 시도 다해보고,

결국 근의공식대입해서 시간초과 최대한 줄일려고 까지 해봤는데, 44%쯤에 있는 하나의 예시에서 탁 막히는 것 같네요..

대체 어디가 문제인걸까요,,

import sys,math
# from sympy import Symbol, solve
a = int(sys.stdin.readline())

i=(3+math.sqrt(9-4*((5-a)/3)))/2 #근의공식 아래의 while문의 식을 이용해 i의 시작값구함
i=int(i)-2 #혹시 모르는 케이스를 위해서 2를뺌
while(1):
    # print(i)
    if i==1:
        ans=i
        break
    if a>1+3*(i-1)*(i-2) and a<1+3*(i)*(i-1):
        ans =i
        break
    i+=1
print(ans)