코딩연습장/백준

[백준/Python] 백준 10813번 공 바꾸기

gayomii 2023. 8. 22. 09:33

🏆백준 10813번 - 공 바꾸기

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

 

10813번: 공 바꾸기

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

www.acmicpc.net


✍️문제

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

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

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


정답

n, m = map(int, input().split())
basket = [i for i in range(1, n+1)]

for _ in range (m):
    i, j = map(int, input().split())
    tempo = basket[i-1]
    basket[i-1] = basket[j-1]
    basket[j-1] = tempo
    
for i in basket:
    print(i, end = " ")

💫 해설 및 한줄평

n개의 바구니와 m번의 반복횟수를 받고, basket에 번호에 해당하는 공을 넣어준다.

for문을 통해 m번의  i와 j를 받고 tempo에 basket의 i-1번째에 해당하는 공을 넣어준다. (basket의 i-1번째는 i번 바구니이다.)

그리고 basket의 i-1번째공 자리에 j-1번째 공을 넣어준다.

j-1번째 공에는 아까 넣어둔 tempo에 들어있는 공을 넣어준다.

마지막으로 basket에 들어있는 공을 end를 통해 공백을 포함해 출력한다.

 

basket을 for문이 들어간 list로 바로 만들어줄때 맨 처음 i를 잊지말기!