[Python] 백준 1547번 - 공

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

---

문제

세준이는 컵 3개를 탁자 위에 일렬로 엎어놓았다. 컵의 번호는 맨 왼쪽 컵부터 순서대로 1번, 2번 3번이고, 세준이는 이 컵을 이용해서 게임을 하려고 한다.

먼저 1번 컵의 아래에 공을 하나 넣는다. 세준이는 두 컵을 고른 다음, 그 위치를 맞바꾸려고 한다. 예를 들어, 고른 컵이 1번과 2번이라면, 1번 컵이 있던 위치에 2번 컵을 이동시키고, 동시에 2번 컵이 있던 위치에 1번 컵을 이동시켜야 한다. 이때 공은 움직이지 않기 때문에, 공의 위치는 맨 처음 1번 컵이 있던 위치와 같다.

세준이는 컵의 위치를 총 M번 바꿀 것이며, 컵의 위치를 바꾼 방법이 입력으로 주어진다. 위치를 M번 바꾼 이후에 공이 들어있는 컵의 번호를 구하는 프로그램을 작성하시오.

 

코드

num = int(input())
cups = [1,2,3]

for i in range(num):
    ch = input().split()
    ch0 = cups.index(int(ch[0]))
    ch1 = cups.index(int(ch[1]))
    cups[ch1], cups[ch0] = cups[ch0], cups[ch1]

print(cups[0])

 

풀이

1. 일단 컵을 바꾼 횟수(num)을 받는다

2. 컵의 위치를 순서대로 리스트(cups)에 저장한다

3. 그리고 컵을 바꾼 횟수만큼 반복해준다

4. 바꾼 컵의 숫자를 입력받고 각각 그 숫자에 해당하는 리스트 인덱스를 찾아 

5. 서로 바꿔준다.

6. 반복문을 끝내고 공이 들어있는 컵의 숫자를 출력한다.