[Kotlin] 알고리즘 - 조합(Combination)

kotlin

private const val alphabets = "ABC"
private val results = mutableListOf<Char>() // 정렬된 알파벳


fun main() {
    combination(2)
}

/**
 * 조합
 * @param count: 추출할 개수
 * @param index: 탐색을 시작할 인덱스. 순서를 고려하지 않으므로 이미 추출된 숫자는 다시 확인할 필요가 없다.
 * @param depth: 추출된 개수
 */
fun combination(count: Int = 0, index: Int = 0, depth: Int = 0) {
    if (depth == count) {
        println("[${results.joinToString(", ")}]") // 완성된 조합
        return
    }

    for (i in index until alphabets.length) {
        results.add(alphabets[i])
        combination(count, i + 1, depth + 1)
        results.removeAt(results.lastIndex)
    }
}

kotlin

fun main() {
    combination(arrayOf("A", "B", "C"), 2).forEach {
        println(it)
    }
}

fun <T> combination(elements: Array<T>, r: Int): List<List<T>> {
    val n = elements.size
    val results = mutableListOf<List<T>>() // 모든 경우의 수

    val result = elements.sliceArray(0 until r)

    fun recursionCombination(depth: Int = 0, index: Int = 0) {
        if (depth == r) {
            results.add(result.toList())
            return
        }

        for (i in index until n) {
            result[depth] = elements[i]
            recursionCombination(depth + 1, i + 1)
        }
    }

    recursionCombination()
    return results
}

text

[A, B]
[A, C]
[B, C]