[구현/수학] 백준 9625 BABBA - Python

 

[ Contents ]

 

 

1. 문제 (링크 참조)

9625번: BABBA

 

 

 

2. 문제 풀이

A -> B -> BA -> BAB -> BABBA

 맨 처음 A가 주어지고, 버튼을 누를 때마다 위와 같은 규칙으로 변환됩니다. 모든 B는 BA로 바뀌고, A는 B로 바뀌는 규칙입니다. K번 눌렀을 때 A와 B의 개수를 구해야 합니다.

 

2022.01.19 - [Algorithm] - [Algorithm] 단골 1번 문제, 구현 / 수학

[Algorithm] 단골 1번 문제, 구현 / 수학

B -> BA
A -> B

 A, B의 개수만 구하면 됩니다. 버튼을 누를 때마다 B만큼 A의 개수가 늘어나고, 누르기 전 A만큼 B의 개수가 늘어납니다.

 

 

 

3. 코드

K = int(input())
a = 1 #맨 처음 A 하나
b = 0
for i in range(K):
    preA = a
    # B -> BA
    a += b
    # A -> B
    a -= preA
    b += preA
print(a, b)

B -> BA로 추가된 A는 B로 변경되면 안됩니다. 누르기 이전의 A만 B로 변경해줍니다.