LeetCode之All Possible Full Binary Trees(Kotlin)

嘟囔發表於2019-01-01

問題: A full binary tree is a binary tree where each node has exactly 0 or 2 children. Return a list of all possible full binary trees with N nodes. Each element of the answer is the root node of one possible tree. Each node of each tree in the answer must have node.val = 0. You may return the final list of trees in any order.

Example 1:

Input: 7 Output: [[0,0,0,null,null,0,0,null,null,0,0],[0,0,0,null,null,0,0,0,0],[0,0,0,0,0,0,0],[0,0,0,0,0,null,null,null,null,0,0],[0,0,0,0,0,null,null,0,0]] Explanation:

LeetCode之All Possible Full Binary Trees(Kotlin)

Note:

  • 1 <= N <= 20

方法: 類似斐波那契數列的原理,先生成1個node的tree,再由1個node的tree生成3個node的tree,直到迭代到N個node的tree即為最終結果,計算過程中通過HashMap儲存生成的tree。

具體實現:

class AllPossibleFullBinaryTrees {
    // Definition for a binary tree node.
    class TreeNode(var `val`: Int = 0) {
        var left: TreeNode? = null
        var right: TreeNode? = null
    }

    private val map = mutableMapOf<Int, List<TreeNode?>>(1 to listOf(TreeNode(0)))

    fun allPossibleFBT(N: Int): List<TreeNode?> {
        if (!map.containsKey(N)) {
            val ans = mutableListOf<TreeNode?>()
            for (x in 0 until N) {
                val y = N - 1 - x
                for (left in allPossibleFBT(x)) {
                    for (right in allPossibleFBT(y)) {
                        val bns = TreeNode(0)
                        bns.left = left
                        bns.right = right
                        ans.add(bns)
                    }
                }
            }
            map.put(N, ans)
        }
        return map[N].orEmpty()
    }
}

fun main(args: Array<String>) {

}
複製程式碼

有問題隨時溝通

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

相關文章