[ BOJ ] 9625 : BABBA ( SILVER 5 ) / Python

문제

상근이는 길을 걷다가 신기한 기계를 발견했다. 기계는 매우 매우 큰 화면과 버튼 하나로 이루어져 있다.

기계를 발견했을 때, 화면에는 A만 표시되어져 있었다. 버튼을 누르니 글자가 B로 변했다. 한 번 더 누르니 BA로 바뀌고, 그 다음에는 BAB, 그리고 BABBA로 바뀌었다. 상근이는 화면의 모든 B는 BA로 바뀌고, A는 B로 바뀐다는 사실을 알게되었다.

버튼을 K번 눌렀을 때, 화면에 A와 B의 개수는 몇 개가 될까?

입력

첫째 줄에 K (1 ≤ K ≤ 45)가 주어진다.

출력

첫째 줄에 A의 개수와 B의 개수를 공백으로 구분해 출력한다.

풀이 과정

한 번 버튼을 누를 때 마다 A의 개수는 그 전의 B의 개수, B의 개수는 그 전의 A의 개수 + 그 전의 B의 개수이다. DP 배열을 이용해 순서대로 계산하여 구한다.

import sys
input = sys.stdin.readline

n = int(input())
dp = [[0, 0]] * 50
dp[0] = [1, 0]

for i in range(1, n+1):
    dp[i] = [dp[i-1][1], sum(dp[i-1])]

print(' '.join(map(str, dp[n])))