P37_2.kt

package ch12

import java.util.*
import java.util.stream.Collectors

class P37_2 {
    fun permute(nums: IntArray): List<List<Int>> {
        // 결과 저장 리스트 선언
        val results: MutableList<List<Int>> = mutableListOf()

        fun dfs(prevElements: MutableList<Int>, elements: List<Int>) {
            // 리프 노드에 도달하면 결과에 추가
            if (elements.isEmpty()) {
                // 자바 8+의 스트림을 이용해 prevElements의 내용을 결과에 삽입
                results.add(prevElements.stream().collect(Collectors.toList()))
            }

            // 전달받은 엘리먼트를 모두 탐색
            for (e in elements) {
                // 전달받은 엘리먼트에서 현재 엘리먼트를 제외하고 nextElements로 새롭게 구성
                val nextElements: MutableList<Int> = ArrayList(elements)
                nextElements.remove(e)

                // prevElements는 기존 값에 현재 엘리먼트 추가
                prevElements.add(e)
                // 재귀 DFS
                dfs(prevElements, nextElements)
                // 돌아온 이후에는 prevElements에서 현재 엘리먼트 삭제
                prevElements.remove(e)
            }
        }

        // 자바 8+의 스트림을 이용해 int[]를 List<Integer>로 변경
        val lst = Arrays.stream(nums).boxed().collect(Collectors.toList())
        // DFS 시작
        dfs(mutableListOf(), lst)
        return results
    }
}