백준 10810 - 공넣기, 10813 - 공 바꾸기

<공 넣기 문제>

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다.
또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 
가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.

도현이는 앞으로 M번 공을 넣으려고 한다. 
도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 
정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다.
만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 
공을 넣을 바구니는 연속되어 있어야 한다.

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

<입력>
첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 
각 방법은 세 정수 i j k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀져 있는 공을 넣는다는 뜻이다. 
예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다. (1 ≤ i ≤ j ≤ N, 1 ≤ k ≤ N)

도현이는 입력으로 주어진 순서대로 공을 넣는다.

<출력>
1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 
공이 들어있지 않은 바구니는 0을 출력한다.

<입력 예시>
5 4
1 2 3
3 4 4
1 4 1
2 2 2
<출력 예시>
1 2 1 1 0

내가 제출한 코드

# 바구니 개수:n, 명령개수:m 받아
n, m = map(int, input().split())

#바구니 번호1부터 시작하기 때문에 n+1개의 리스트 생성
#리스트 인덱스와 바구니 번호 일치시키기 위해 
list = [0] * (n+1)


#i부터 j까지의 인덱스에 k값을 넣어줌 
for __ in range(m):
    i, j, k = map(int, input().split())
    for idx in range(i, j+1):
        list[idx] = k

#리스트 모든 원소 출력, 0번 바구니는 출력하지 않아야 하므로 1부터 n+1까지 출력
#end를 사용해서 값 사이 공백 넣기
for i in range(1, n+1):
    print(list[i], end=' ')

 

 

<공 바꾸기 문제>

도현이는 바구니를 총 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번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다.

<예제 입력>
5 4
1 2
3 4
1 4
2 2
<예제 출력>
3 1 4 2 5

제출 코드

# 바구니 개수:n, 명령개수:m
n, m = map(int, input().split())

# 바구니 번호1부터 시작하기 때문에 n+1개의 리스트 생성
# 리스트 인덱스와 바구니 번호 일치시키기 위해
list = [0] * (n+1)

# i부터 n+1까지의 인덱스에 j값을 넣어줌
# 바구니번호 = 공번호 일치 시키기
for j in range(1, n+1):
    list[j] += j

# 공번호 교환하기 temp를 임시적인 값 저장에 활용
for _ in range(1, m+1):
    a, b = map(int, input().split())
    temp = list[a]
    list[a] = list[b]
    list[b] = temp

# 리스트 모든 원소 출력, 0번 바구니는 출력하지 않아야 하므로 1부터 n+1까지 출력
# end를 사용해서 값 사이 공백 넣기
for i in range(1, n+1):
    print(list[i], end=' ')