백준 알고리즘 1309. 동물원

https://www.acmicpc.net/problem/1309

1309번: 동물원

 

전체 코드

N = int(input())


def solution(N):
    lion = 2
    empty = 1
    for i in range(2, N + 1):
        lion, empty = (lion + 2 * empty) % mod, (lion + empty) % mod

    return (lion + empty) % mod


print(solution(N))