[BOJ#9625] BABBA

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

9625번: BABBA

1 초

128 MB

12408

8071

7275

65.470%

문제

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

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

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

입력

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

출력

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

---

제출 코드1

# 함수를 만들고 각각 정의해준다.
# 그리고 리턴값으로 다시 그 함수를 받는다.
# A - B - BA - BAB - BABBA - BABBABAB - BABBABABBABBA
# BABBABABBABBABABBABAB - BABBABABBABBABABBABABBABBABABBABBA
#        0   1    2   3   4   5   6   7   8   9    10 
# 규칙 A: 1 - 0 - 1 - 1 - 2 - 3 - 5 - 8 - 13 - 21 - 34
# 규칙 B: 0 - 1 - 1 - 2 - 3 - 5 - 8 - 13 - 21 - 34 - 55

K = int(input())
dp = [0] * (K+1)
dp[0] = [1, 0]

for cnt in range(1, K+1):
    dp[cnt] = dp[cnt-1][1], dp[cnt-1][0] + dp[cnt-1][1]

print(*dp[K])

 

규칙을 찾아서 리스트형태로 넣어준 코드 ( DP 이해에 스터디 친구들의 도움을 많이 받았다.. 무한감사!)

제출 코드2

K = int(input())
dp = [0] * (K+2)
dp[0] = 1
dp[1] = 0

for i in range(2, K+2):
    dp[i] = dp[i-1] + dp[i-2]

print(f'{dp[K]} {dp[K+1]}')

 

DP에 대해 조금 이해하고 생각하던 구현완성