[白俊]1769:3の倍数

質問する

タイムアウト解

import sys

n = sys.stdin.readline()
answer = 0
count = 0

def sol(n):
    global answer, count
    if len(n) == 1 :
        if n in [3, 6, 9]:
            answer = "YES"
        else:
            answer = "NO"
        return
    
    count += 1
    a = 0 # 각 자리 숫자의 합
    for i in range(len(n)):
        a += int(n[i])
    
    sol(a)

sol(n)
print(count)
print(answer)

正解を解く

import sys

n = sys.stdin.readline().strip()
answer = 0
count = 0

def sol(n):
    global answer, count
    if len(n) == 1 :
        if n in ['3', '6', '9']:
            answer = "YES"
        else:
            answer = "NO"
        return
    
    count += 1
    a = 0 # 각 자리 숫자의 합
    for i in range(len(n)):
        a += int(n[i])
    
    sol(str(a))

sol(n)
print(count)
print(answer)