[백준] 10813번 파이썬 python(공 바꾸기)

▷ 문제

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

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

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

 

▷ 전체 코드

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

X = []

for i in range(N):
    X.append(i+1)

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

for _ in range(N):
    print(X[_], end = " ")

 

▷ 코드 분석

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

변수 N에 바구니의 개수를 입력한다. 변수 M에 공을 몇 번 바꿀건지 입력한다.

 

map() 함수를 사용해 입력한 요소에 지정된 함수를 적용한다.

input() 함수를 사용해 입력 받은 요소를 문자열로 변수에 저장한다. split() 함수를 사용해 입력 받은 요소를 공백을 기준으로 나눠준다.

int() 함수를 사용해 문자열을 정수형으로 바꿔준다.

 

X = []

리스트 X를 만든다.

 

for i in range(N):
    X.append(i+1)

각 번호에 해당하는 공이 들어있는 N개의 바구니를 만든다.

 

for 문을 사용해 N번 반복하는 반복문을 만든다.

.append() 함수를 사용해 리스트 X에 1번부터 N번까지의 공이 있는 N개의 바구니를 추가한다.

 

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

i번 바구니와 j번 바구니에 들어있는 공을 교환하는 것을 M번 반복한다.

 

for 문을 사용해 M번 반복하는 반복문을 만든다.변수 i와 j에 몇 번째 바구니의 공을 교환할지 공백을 기준으로 입력한다.리스트 X의 i번째 값과 j번째 값을 교환한다.

 

 

for _ in range(N):
    print(X[_], end = " ")

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

 

for 문을 사용해 N번 반복하는 반복문을 만든다.

리스트 X에 있는 각 요소들을 공백으로 나눠서 출력한다.