백준 10813번 - 공 바꾸기 (Python 문제 풀이 및 해설)

문제

 

예제

 

문제 풀이

해당 문제의 핵심은 배열 교환(swap)하는 것입니다. 교환할 값 2개 a, b 중 a를 임시로 만든 변수(t)에 담고, a에 b를 담고 b에 t를 할당하면 swap이 됩니다.

 

우선 모든 바구니에 입력받은 값을 채워 넣습니다. 예를들어, n이 5인 경우 [1, 2, 3, 4, 5] 와 같이 담습니다.

이후 i값과 j값을 입력받고 2개의 원소를 swap 합니다. 이때 주의할 점은 0번째 인덱스에 1번째 바구니 값을 할당 했으므로, 입력받은 값에 각각 -1을 해줘야합니다

 

import sys

n, m = map(int, sys.stdin.readline().split())

bucket = [i + 1 for i in range(n)]

for _ in range(m):
    i, j = map(lambda x: int(x) - 1, sys.stdin.readline().split())
    bucket[i], bucket[j] = bucket[j], bucket[i]

print(*bucket)

 

파이썬에서 스왑을 할때는 a, b = b, a와 같이 할 수있습니다.

 

채점 결과

 

원본 코드

https://github.com/bluemiv/Algorithm/blob/master/baekjoon/ex10813/ex10813.py

 

관련 문제

비슷한 문제로 공 넣기 문제도 있으니 참고바랍니다.

백준 10810번 - 공 넣기 (Python 문제 풀이 및 해설)