[백준/python][구현] 10812번 - 바구니 순서 바꾸기

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

10812번: 바구니 순서 바꾸기

제한조건

 

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 순서대로 적혀 있다. 바구니는 일렬로 놓여 있고, 가장 왼쪽 바구니를 1번째 바구니, 그다음 바구니를 2번째 바구니,..., 가장 오른쪽 바구니를 N번째 바구니라고 부른다.

도현이는 앞으로 M번 바구니의 순서를 회전시키려고 만들려고 한다. 도현이는 바구니의 순서를 회전시킬 때, 순서를 회전시킬 범위를 정하고, 그 범위 안에서 기준이 될 바구니를 선택한다. 도현이가 선택한 바구니의 범위가 begin, end이고, 기준이 되는 바구니를 mid라고 했을 때, begin, begin+1,..., mid-1, mid, mid+1,..., end-1, end 순서로 되어있는 바구니의 순서를 mid, mid+1,..., end-1, end, begin, begin+1,..., mid-1로 바꾸게 된다.

바구니의 순서를 어떻게 회전시킬지 주어졌을 때, M번 바구니의 순서를 회전시킨 다음, 바구니에 적혀있는 번호를 가장 왼쪽 바구니부터 출력하는 프로그램을 작성하시오.

---

입력

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

둘째 줄부터 M개의 줄에는 바구니의 순서를 바꾸는 만드는 방법이 주어진다. 방법은 i, j, k로 나타내고, 왼쪽으로부터 i번째 바구니부터 j번째 바구니의 순서를 회전시키는데, 그때 기준 바구니는 k번째 바구니라는 뜻이다. (1 ≤ i ≤ k ≤ j ≤ N)

도현이는 입력으로 주어진 순서대로 바구니의 순서를 회전시킨다.

---

출력

모든 순서를 회전시킨 다음에, 가장 왼쪽에 있는 바구니부터 바구니에 적혀있는 순서를 공백으로 구분해 출력한다.

---

예제 입력 1	예제 출력 1
10 5 1 6 4 3 9 8 2 10 5 1 3 3 2 6 2	1 4 6 2 3 7 10 5 8 9

 

코드

from sys import stdin  
r_line = stdin.readline

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

# begin, mid, end
# i    j      k
# 시작 끝     기준
arr = [str(_) for _ in range(N+1)]
temp_arr = [str(_) for _ in range(N+1)]

for i in range(1, M+1): # M번만큼 반복
    b, e, m = map(int, r_line().split())

    for j in range(b, m): # 임시배열저장
        temp_arr[e-m+b+1 +j-b] = arr[j] # e-m+b+1:초기값, j-b:증감수

    for j in range(m, e+1): # mid ~ end
        arr[b+(j-m)] = arr[j]

    cnt = e - m + b + 1
    for k in range(cnt, e+1): # begin ~ mid-1
        arr[k] = temp_arr[k]

print(' '.join(arr[1:]))

풀이

for문을 총 3단계로 나누었다.

1) 첫 번째 for문: 요소값들을 임시로 저장할 배열

2) 두 번째 for문: mid ~ end까지 회전값 대입

3) 세 번째 for문: begin ~ (mid-1)까지 회전값 대입

 

배열을 1번부터 받기 위해 1칸 리스트를 더 받았기 때문에

join으로 출력하되 1번 idx부터 출력한다.