글 읽기 - 시간초과를 해결하지 못하겠습니다.

수학적으로 다소 헷갈리지만 열심히 등차수열을 공부해서 풀었는데

시간 초과가 뜹니다

혹시 어떻게 해야 시간을 줄일수 있을까요?

a=int(input())
i = 0

while True:
    if 3*i*(i-1)+2 <= a <= 3*i*(i+1)+1:
        break
    else:
        i += 1

print(i+1)

a가 1인경우 무한루프가 돌아 시간초과가 뜹니다.

댓글을 작성하려면 로그인해야 합니다.

Baekjoon Online Judge