讨论/算法和数据结构/在二叉树中增加一行/
在二叉树中增加一行

给定一个二叉树,根节点为第1层,深度为 1。在其第 d 层追加一行值为 v 的节点。

添加规则:给定一个深度值 d (正整数),针对深度为 d-1 层的每一非空节点 N,为 N 创建两个值为 v 的左子树和右子树。

将 N 原先的左子树,连接为新节点 v 的左子树;将 N 原先的右子树,连接为新节点 v 的右子树。

如果 d 的值为 1,深度 d - 1 不存在,则创建一个新的根节点 v,原先的整棵树将作为 v 的左子树。

示例 1:

输入: 
二叉树如下所示:
       4
     /   \
    2     6
   / \   / 
  3   1 5   

v = 1

d = 2

输出: 
       4
      / \
     1   1
    /     \
   2       6
  / \     / 
 3   1   5  

示例 2:

输入: 
二叉树如下所示:
      4
     /   
    2    
   / \   
  3   1    

v = 1

d = 3

输出: 
      4
     /   
    2
   / \    
  1   1
 /     \  
3       1

注意:

  1. 输入的深度值 d 的范围是:[1,二叉树最大深度 + 1]。
  2. 输入的二叉树至少有一个节点。
展开讨论

前置遍历找到需要插入层的上一层所有节点,然后插入对应节点。

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} v
 * @param {number} d
 * @return {TreeNode}
 */
var addOneRow = function (root, v, d) {
    function findDNodes(root, d) {
        let resultNodes = [root];
        for (let i = 1; i < d; i++) {
            resultNodes = resultNodes
                .filter(node => node)
                .reduce((sofar, curr) => {
                    return sofar.concat([curr.left, curr.right]);
                }, []);
        }
        return resultNodes;
    }

    if (d === 1) {
        const insertNode = new TreeNode(v);
        insertNode.left = root;
        return insertNode;
    }

    const lastNodes = findDNodes(root, d - 1);
    lastNodes
        .filter(node => node)
        .forEach(node => {
            if (node) {
                const insertNodeL = new TreeNode(v);
                const insertNodeR = new TreeNode(v);
                insertNodeL.left = node.left;
                insertNodeR.right = node.right;
                node.left = insertNodeL;
                node.right = insertNodeR;
            }
        })
    return root;
};
展开全部 3 讨论