[Python] 백준 10813번 공 바꾸기

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다.

바구니에는 공이 1개씩 들어있고, 처음에는 바구니에 적혀있는 번호와 같은 번호가 적힌 공이 들어있다.

도현이는 앞으로 M번 공을 바꾸려고 한다.

도현이는 공을 바꿀 바구니 2개를 선택하고, 두 바구니에 들어있는 공을 서로 교환한다.

공을 어떻게 바꿀지가 주어졌을 때,

M번 공을 바꾼 이후에 각 바구니에 어떤 공이 들어있는지 구하는 프로그램을 작성하시오.

 

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

10813번: 공 바꾸기

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 교환할 방법이 주어진다.

각 방법은 두 정수 i j로 이루어져 있으며, i번 바구니와 j번 바구니에 들어있는 공을 교환한다는 뜻이다. (1 ≤ i ≤ j ≤ N)

도현이는 입력으로 주어진 순서대로 공을 교환한다.

출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다.

예제 입력

5 4
1 2
3 4
1 4
2 2

예제 출력

3 1 4 2 5

문제 풀이

N,M = map(int,input().split())

L = list(range(1,N+1))
temp = 0

for _ in range(M):
    i,j = map(int,input().split())
    temp = L[i-1]
    L[i-1] = L[j-1]
    L[j-1] = temp

print(*L)

지난번 10810번 문제와는 다르게 두 바구니의 숫자를 변경하는 문제입니다.

이를 위해 temp 변수를 활용하여 3가지를 서로 변경하였습니다.

 

1) temp 변수 선언

2) i번째 바구니의 값을 temp에 넣기

3) j번째 바구니의 값을 i에 넣기

4) temp의 값을 j번째 바구니에 넣기

 

만약 이러한 과정을 거치지 않고, 

i의 값을 j에 넣고 j의 값을 i에 넣었다면 아래와 같은 오류가 발생했을 것입니다.

i = 2
j = 4

i = j # i = 4, j = 4
j = i # i = 4, j = 4

 

그러나 파이썬은 위와 같은 과정을 거치지 않고 두 값을 서로 바꿀 수도 있습니다.

N,M = map(int,input().split())
L = list(range(1,N+1))

for _ in range(M):
    i,j = map(int,input().split())
    L[i-1], L[j-1] = L[j-1], L[i-1]

print(*L)

 

리스트를 한 줄로 출력하는 방법은 아래 링크에서 추가 확인 가능합니다.

 

https://jyzinn.tistory.com/24

[Python] BOJ 10810번 공 넣기

---

저처럼 파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.