옹실이의 개발이야기

알고리즘

[백준] 알고리즘 #10813 : 공 바꾸기 (Python)

옹실 2023. 5. 5. 23:32

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 바구니에는 공이 1개씩 들어있고, 처음에는 바구니에 적혀있는 번호와 같은 번호가 적힌 공이 들어있다.

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

공을 어떻게 바꿀지가 주어졌을 때, M번 공을 바꾼 이후에 각 바구니에 어떤 공이 들어있는지 구하는 프로그램을 작성하시오.

 

입력

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

둘째 줄부터 M개의 줄에 걸쳐서 공을 교환할 방법이 주어진다. 각 방법은 두 정수 i j로 이루어져 있으며, i번 바구니와 j번 바구니에 들어있는 공을 교환한다는 뜻이다. (1 ≤ i ≤ j ≤ N)

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

 

출력

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

 

풀이

1. 바구니 개수를 n, 공을 교환할 방법을 받을 회수를 m에 담는다.
2. n개만큼 공을 담을 바구니 list(배열 b)를 만들고,
   배열 인덱스번호를 그대로 담는다.
   바구니는 1번부터 시작하므로, 배열도 1번부터 n+1개만큼 만든다.  
3. m만큼 for문 돌려 공을 교환할 방법을 입력받아 i, j에 담는다.
4.  for문이 한번씩 돌 때마다 배열 i번째 있는 값과 j번째에 있는 값을 바꿔서 세팅한다.
     ※ i에 j값을 덮어씌운 후, i에 있는 값을 다시 j에 덮어씌우면 원래 i값이 j값으로 바뀐 상태이므로, 
         값을 바꾸기 전에 i번째 있는 값을 x에 따로 담아놓고 값을 바꿨다.

1
2
3
4
5
6
7
8
9
10
n, m = map(int, input().split())
= [c for c in range(1, n+1)] #배열 인덱스 번호 담음
 
for x in range(m):
    i, j = map(int, input().split())
    x = b[i-1]
    b[i-1= b[j-1]
    b[j-1= x
 
print(*b) #배열 전체 출력
cs