BeakJoon - 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)

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

 

출력

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

 

 

---

문제 분석

 

• 바구니의 범위 : begin, end
• 기준점 : mid

 

일단 바구니의 범위는 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 이런식으로 바꾼다고 한다.

 

먼저 그림을 그려보았다.

 

 

자세히 보니 mid의 기점으로 begin에서 mid-1, mid에서 end가 temp되는 것이었다.

 

 

입력을 분석해보면

 

첫째 줄

• N은 N개의 바구니를 1부터 N까지 생성한다.
• M은 바구니의 회전하는 횟수이다.

 

둘째 줄 ~

• i, j, k개 주어지는데
• i와 j는 begin과 end, 즉 바구니의 범위
• k는 기준점이다.

 

그림으로 그려보겠다.

 

 

문제를 해석한 것과 같이 이렇게 바뀐다.

 

 

코드 구현

 

위 문제를 파이썬으로 구현하였다.

 

n, m = input().split()
n, m = int(n), int(m)

array = []

for q in range(1,n+1):
    array.append(q)

for a in range(m):
    i, j, k = input().split()
    begin, end, mid = int(i)-1, int(j)-1, int(k)-1
    temp = array[mid:end+1] + array[begin:mid]
    array = array[:begin] + temp + array[end+1:]

for i in array:
    print(i,end=' ')

 

코드 풀이

 

코드를 하나하나 풀이 해보면

 

n, m = input().split()
n, m = int(n), int(m)

 

바구니의 갯수n과 회전횟수 m을 받아서 정수로 변환한다.

 

array = []

for q in range(1,n+1):
    array.append(q)

 

배열 선언 후 array에 바구니의 입력 갯수만큼 1부터 차근차근 넣어준다.

 

for a in range(m):
    i, j, k = input().split()
    begin, end, mid = int(i)-1, int(j)-1, int(k)-1
    temp = array[mid:end+1] + array[begin:mid]
    array = array[:begin] + temp + array[end+1:]

 

루프는 m의 회전 횟수만큼 돌리며 i, j, k는 begin, end, mid 이다. index로 구분하기 위해 각각 1씩 빼주었다.

그리고 temp 변수에 mid에서 end, begin에서 mid-1을 인덱싱해 자리를 교체하였다. 인덱싱 할때 [시작: 끝-1] 이기때문에 각각의 값을 [mid: end+1], [begin: mid]로 조정해주었다.

 

그리고 array를 재정의 해주었는데 temp양쪽에 있는 array[ : begin]과 array[end+1 : ]은 다음 그림과 같다. 

 

 

 

만약 begin, end의 범위가 전체 범위라면 괜찮겠지만, 범위가 중간이나 한 쪽으로 취우쳐져있으면 temp만으로 array를 재정의를 할 수 없기 때문이다.

 

for i in array:
    print(i,end=' ')

 

마지막으로 공백을 기준으로 출력하였다.