백준 1769번 - 3의 배수 - Python

쉽게 풀릴 것 같던 문제였지만, 시간 초과에 빠져버렸다.. 이런.. 아마도 각 자리 숫자를 더하는 과정에서 시간이 오래걸리는 듯하다.

PS에서 시간 초과가 가장 답답한 듯하다. 터미널에서는 순식간에 구해지는 것들이 사이트 상에서는 느리다고 하니까. 일단 Python에서 Pypy로 바꾸어 제출했고, while문에서 len(n) == 1처럼 len()을 이용하여 한자릿수인지 체크를 했다. int(n)<10보다 len()이 시간적으로 효율적이다.

n = input()
count = 0

while True:
    if len(n) == 1:
        break

    count += 1
    temp = 0
    for i in range(len(n)):
        temp += int(n[i])
    n = str(temp)

    if len(n) == 1:
        break

print(count)
if n in ['3','6','9']:
    print("YES")
else:
    print("NO")

Comments: