[구현/수학] 백준 20361 일우는 야바위꾼 - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

20361번: 일우는 야바위꾼

 

 

 

2. 문제 풀이

 야바위를 직접 구현하는 문제입니다.

 

 

3. 코드

import sys
input = sys.stdin.readline

N, X, K = map(int, input().split())
cup = [0] * (N+1)
cup[X] = 1
for _ in range(K):
    a, b = map(int, input().split())
    cup[a], cup[b] = cup[b], cup[a]

for idx, i in enumerate(cup):
    if i == 1:
        print(idx)

 입력에서 주어지는 대로 둘을 swap합니다. 그리고 맨 마지막에 공이 어디있는지를 출력합니다.