[백준] 20361번 일우는 야바위꾼 (파이썬)

 

20361번: 일우는 야바위꾼

 

문제

전설의 야바위꾼 일우는 Shell Game으로 야바위를 한다. Shell Game은 다음과 같은 절차로 진행된다.

1. 진행자가 N개의 컵을 일렬로 놓고, 그 중 X번째 컵에 공을 숨겨둔다.
2. 임의의 서로 다른 두 컵의 위치를 맞바꾼다. 이 항목을 K번 수행한다. 만약, 공을 숨겨둔 컵을 움직인다면 공도 그 컵을 따라서 움직인다.
3. 참가자는 몇 번째 컵에 공이 숨겨져 있는지 추측한다.
4. 그 컵에 공이 숨겨져 있다면 참가자가, 그렇지 않다면 진행자가 이긴다.

수혁이는 Shell Game을 잘하고 싶다. 하지만, 일우가 진행자라면 무슨 수를 써도 이길 수 없어 수혁이는 일우의 사기도박을 의심하고 있다. 현재 우리는 수혁과 일우가 진행한 Shell Game의 모든 기록을 입수했다. 이를 바탕으로, 일우가 사기도박을 하지 않았다면 공이 몇 번째 컵에 있는지 알려주는 프로그램을 작성하자.

입력

첫째 줄에 N (3 ≤ N ≤ 200,000), X (1 ≤ X ≤ N), K (1 ≤ K ≤ 100,000)가 공백으로 구분되어 주어진다.

둘째 줄부터 K개의 줄에는 순서대로 바꾼 두 컵의 위치 Ai, Bi (1 ≤ Ai, Bi ≤ N, Ai ≠ Bi)가 공백으로 구분되어 주어진다.

주어지는 모든 입력은 정수다. 

3 2 4
1 3
3 2
3 1
2 3

출력

일우가 사기도박을 하지 않았다면, 공이 몇 번째 위치의 컵에 있어야 하는지 정수로 출력하시오. 

---

코드

n, x, k = map(int, input().split())
arr = []
for i in range(0, n):
    arr.append(i + 1)
change = [list(map(int, input().split())) for _ in range(k)]

for i in range(k):
    arr[change[i][0] - 1], arr[change[i][1] - 1] = arr[change[i][1] - 1], arr[change[i][0] - 1]

print(arr.index(x) + 1)

 

풀이

현재 컵을 arr 리스트로 나타내고, change 리스트에 바꿀 컵 위치들을 입력받는다.

처음 컵이 1, 2, 3 순서대로 있을 것이고, 이를 1과 3의 위치를 바꾼다면 arr[1] ↔ arr[3]이 되도록 구현하였다. 

입력받은 대로  모두 위치를 바꾸고 나면, 원래 공이 있던 x의 인덱스를 찾아 출력시켜준다.