백준 10813 공 바꾸기/파이썬

📌문제

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

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

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

 

📌풀이

바구니 리스트에 바구니 번호와 같은 번호로 공의 번호를 초기화

공을 바꿀 두 바구니 번호(A, B)를 입력받고 바구니 A의 값을 임시로 저장해 둔 후 다른 바구니 B의 값을 A에 저장

이후 바구니 B에 임시로 저장해 둔 바구니 A의 값을 저장해서 리스트의 값을 바꾼다

 

📌코드

n, m = map(int, input().split())
baskets = [0] * n

# 바구니 번호와 같은 공의 번호로 초기화
for q in range(n):
    baskets[q] = q + 1

for w in range(m):
    i, j = map(int, input().split())
    temp = baskets[i - 1]
    baskets[i - 1] = baskets[j - 1]
    baskets[j - 1] = temp

for e in range(n):
    print(baskets[e], end=" ")