IT/ps

백준 8394번 약수

u149_cinderella 2025. 3. 8. 00:02

 

피보나치와 같은 점화식이다.

import sys

n=int(sys.stdin.readline().rstrip())
if n==1:
    print(1)
elif n==2:
    print(2)
else:
    prev=2
    prev_prev=1
    result=0
    for i in range(3,n+1):
        result=(prev+prev_prev)%10
        prev_prev=prev
        prev=result
    print(result)

다만 변수로 저장할때 2개만 써야하고 마지막 자리만 출력하는걸 조심해야한다