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

문제

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

  • 배열의 아이템들 위치를 서로 바꾸는 단순한 배열 활용 문제

풀이

주어진 input들을 모두 받아서 배열을 초기화해주고 temp 변수를 활용하여 반복문을 돌며 자리를 변경해야하는 값들을 변경해주었다. 결과값 배열 출력에는 *를 활용하였다.(Unpacking을 해주는 역할을 한다)

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

# init list
ls = [ball+1 for ball in range(N)]

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

print(*ls)

후기

파이썬 경험이 부족하여 기초 문제들을 조금씩 풀고 있는데, 몸풀기용으로 적절한 문제였다. 리스트 컴프리헨션도 꾸준히 사용하다보니 손에 익은 것 같다.