【路飞】95. 不同的二叉搜索树 II js实现

91 阅读1分钟

上题目:

给你一个整数 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));
    }
};