[ 백준 파이썬 문제풀이 ] 단계별로 풀어보기 / 4-6 1차원 배열 공 바꾸기 #10813

[ 백준 파이썬 문제풀이 ] 단계별로 풀어보기 / 4-6 1차원 배열  공 바꾸기 #10813

 

---

 

문제 📢

도현이는 바구니를 총 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 2 3 4 5

2 1 3 4 5

2 1 4 3 5

3 1 4 2 5

 

내가 적은 답안 💡

import sys

N, M = map(int, sys.stdin.readline().split())
basket = list(range(1, N+1))	#basket이란 리스트 생성 후 1부터 N+1까지의 요소를 저장


for i in range(M):
    try:
        A, B = map(int, sys.stdin.readline().split())
        basket[A-1], basket[B-1] = basket[B-1], basket[A-1]	#A-1과 B-1번재에 있는 요소를 서로 바꿈
    except:
        break
        
print(' '.join(map(str, basket)))

 

basket = list(range(1, N+1)) 에서 범위를 1부터 N+1 한 까닭과

basket[A-1], basket[B-1] = basket[B-1], basket[A-1] 에서 각 입력받은 정수에서 -1를 한 까닭은

 

리스트 함수에서는 0번째 부터 시작하기 때문이다

 

아래 링크는 리스트에 관련된 함수들을 정리한 사이트다. 한 번씩 확인해보면 좋을 것 같다

 

https://docs.python.org/ko/3.13/tutorial/datastructures.html

5. Data Structures

 

 

느낀점🎻

 

저번 공 넣기 문제를 풀어보니 다른 문제들이 쉬워 보인다

 

---

 

이 블로그는 개발자를 꿈꾸는 사람이 열심히 공부하며 적어가는 공간입니다.

제 답안과 풀이가 다를 수 있지만, 정확한 정보를 위해 여러번 검토하며 적겠습니다!