問題: 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:
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>) {
}
複製程式碼
有問題隨時溝通