Kotlin 🍬 백준 6단계 :: 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)

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

 

출력   |

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

 

 

풀이  |

  k (mid)를 기준으로 i (begin) 부터 k - 1, k + 1 부터 j (end)를 바꿔준다.

  즉, j - k + 1 (== k - i + 1) 반복하여 맨 뒤에 위치한 요소를 앞으로 이동시켜준다.

  간결한 코드를 위해 위치를 바꿔주는 함수 swap()을 array의 확장함수로 생성해 구현해주었다.

 

 

답안  |

import java.util.Scanner

fun main() = with(Scanner(System.`in`)) {
    val N = nextInt()
    val M = nextInt()
    var buckets = Array(N) { it + 1 }
    
    repeat(M) {
        val i = nextInt()
        val j = nextInt()
        val k = nextInt()
        
        repeat(j - k + 1){
            for (num in j - 1 downTo i){
                buckets.swap(num, num - 1)
            }
        }
    }
    
    buckets.forEach { print("${it} ") }
}

fun Array<Int>.swap(n: Int, m: Int){
    val temp = this[n]
    this[n] = this[m]
    this[m] = temp
}