上题目:
给你一个整数 n ,请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。
示例 1:
输入:n = 3 输出:[[1,null,2,null,3],[1,null,3,2],[2,1,3],[3,1,null,null,2],[3,2,null,1]] 示例 2:
输入:n = 1 输出:[[1]]
提示:
1 <= n <= 8
题解:
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {number} n
* @return {TreeNode[]}
*/
var generateTrees = function(n) {
const d = [];
if (n === 0) return d;
d[1] = [new TreeNode(1)];
if (n === 1) return d[1];
for (let i=2; i<=n; i++) {
let res = [];
for (let j=0;j<d[i-1].length; j++) {
let node = d[i-1][j];
let p = node;
let stack = [];
while(p) { //将所有二叉树右半部分所有的右节点入栈
stack.push(p);
p = p.right;
}
let father;
while(stack.length > 0) {
addNode = new TreeNode(i);
p = stack.pop();
father = stack[stack.length - 1];
if (!p.right) { //如果是叶子节点
p.right = addNode;
res.push(deepCopy(node));
p.right = null; //这里要还原,以免改变了原来树的结构
}
addNode.left = p; //将当前右节点作为新节点的左子树
if (father) { //如果不是根节点
father.right = addNode; //替换当前的右节点
res.push(deepCopy(node));
father.right = p; //这里要还原,以免改变了原来树的结构
} else { //如果是根节点
res.push(addNode);
}
}
}
d[i] = res;
}
return d[n];
function TreeNode(val, left, right) {
this.val = (val===undefined ? 0 : val)
this.left = (left===undefined ? null : left)
this.right = (right===undefined ? null : right)
}
function deepCopy(node) {
return JSON.parse(JSON.stringify(node));
}
};