LeetCode之All Paths From Source to Target(Kotlin)

嘟囔發表於2019-01-01

問題: Given a directed, acyclic graph of N nodes. Find all possible paths from node 0 to node N-1, and return them in any order. The graph is given as follows: the nodes are 0, 1, ..., graph.length - 1. graph[i] is a list of all nodes j for which the edge (i, j) exists.

複製程式碼

Example: Input: [[1,2], [3], [3], []] Output: [[0,1,3],[0,2,3]] Explanation: The graph looks like this: 0--->1 | | v v 2--->3 There are two paths: 0 -> 1 -> 3 and 0 -> 2 -> 3.

複製程式碼

方法: 深度優先遍歷,按照陣列中儲存的路徑遍歷,遍歷所有的path即可得到結果。

具體實現:

class AllPathsFromSourceToTarget {
    fun allPathsSourceTarget(graph: Array<IntArray>): List<List<Int>> {
        val result = mutableListOf<List<Int>>()
        val path = mutableListOf<Int>()
        path.add(0)
        dfsSearch(graph, path, result, 0)
        path.remove(0)
        return result
    }

    private fun dfsSearch(graph: Array<IntArray>, path: MutableList<Int>, result: MutableList<List<Int>>, node: Int) {
        if (node == graph.lastIndex) {
            result.add(ArrayList<Int>(path))
            return
        }
        for (nextNode in graph[node]) {
            path.add(nextNode)
            dfsSearch(graph, path, result, nextNode)
            path.remove(nextNode)
        }
    }
}

fun main(args: Array<String>) {
    val arrays = arrayOf(intArrayOf(1,2), intArrayOf(3), intArrayOf(3), intArrayOf())
    val allPathsFromSourceToTarget = AllPathsFromSourceToTarget()
    val result = allPathsFromSourceToTarget.allPathsSourceTarget(arrays)
}
複製程式碼

有問題隨時溝通

具體程式碼實現可以參考Github

相關文章