[구현/수학] 백준 10813 공 바꾸기 - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

10813번: 공 바꾸기

 

 

2. 문제 풀이

 N개의 공이 나열되어 있고, 이를 M번 Swap하는 문제입니다.

 

 

3. 코드

import sys
input = sys.stdin.readline

N, M = map(int, input().split())
lst = list(range(1, N+1))
for _ in range(M):
    i, j = map(int, input().split())
    lst[i-1], lst[j-1] = lst[j-1], lst[i-1]
print(" ".join(list(map(str, lst))))

 기존 프로그래밍 언어는 tmp 변수를 통해 swap을 해야하지만, 파이썬은 한 줄 코드로 가능합니다.

 a, b = b, a로도 swap이 됩니다.