[백준/Python] 10812번 : 바구니

시간 제한	메모리 제한	문제 티어	정답 비율
1 초	256 MB	브론즈 II	63.165%

📜 문제

도현이는 바구니를 총 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)

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

 

 

📤출력

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

 

💡풀이

이 문제의 핵심은 "리스트의 인덱스와 슬라이스 기능을 잘 활용하였는가?" 였다.

 

리스트의 인덱스는 0부터 시작이 되고, list_name[a:b]는 a부터 b-1까지의 요소들이 들어 있는 리스트를 출력한다.

 

문제에서는 바구니 순서를 1부터 시작하므로, 이렇게 종합을 하면 0부터 i-2까지 요소 + k-1부터 j-1까지 요소 + i-1부터 k까지 요소 + j부터 끝까지의 요소를 합치면 된다.

 

나는 이러한 문제의 경우 잘 생각이 안날 때는 종이에다가 직접 시뮬레이션해서 해당 인덱스가 어디인지 찾아보는 편이다.

 

import sys

input = lambda: sys.stdin.readline().rstrip()

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

basket_list = [i for i in range(1, N+1)]

for a in range(M):
  i, j, k = map(int, input().split())
  basket_list = basket_list[:i-1] + basket_list[k-1:j] + basket_list[i-1:k-1] + basket_list[j:]

print(*basket_list)