Baekjoon(Kotlin) - 15649. N과 M (1)

URL : https://www.acmicpc.net/problem/15649

15649번: N과 M (1)

 

[ RESULT ]

val br = System.`in`.bufferedReader()
val bw = System.`out`.bufferedWriter()

fun main() {

    val (N, M) = br.readLine().split(" ").map { it.toInt() }

    val arr = Array(N) { it + 1 }

    permutation(arr, N, M).forEach {
        bw.write(it.joinToString(" ") + "\n")
    }

    br.close()
    bw.close()
}

fun <T> permutation(Arr: Array<T>, N: Int, R: Int): List<List<T>> {
    val nArray = Arr.sliceArray(0 until R)
    val visited = BooleanArray(N)
    val list = mutableListOf<List<T>>()

    fun recursionPermutation(depth: Int = 0) {
        if (depth == R) {
            list.add(nArray.toList())
            return
        }

        Arr.forEachIndexed { index, t ->
            if(!visited[index]) {
                visited[index] = true
                nArray[depth] = t
                recursionPermutation(depth + 1)
                visited[index] = false
            }
        }
    }

    recursionPermutation()
    return list
}

 

 

BackTracking : 모든 경우의 수를 확인하는 것을 말함

 

효율적인 순열 코드를 구상하던 중 발견한 블로그

https://gyeong-log.tistory.com/51

[코틀린] 조합, 순열 구현하기