Leetcode623. 在二叉树中增加一行

Leetcode623. 在二叉树中增加一行

相似题目：
Leetcode104. 二叉树的最大深度
题目：
给定一个二叉树，根节点为第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

题解：
计算每一层的深度，使用迭代。
java代码：

/**
     * @param root
     * @param v
     * @param d
     * @return
     */
    public static TreeNode addOneRow(TreeNode root, int v, int d) {
        if (d == 1) {
            TreeNode node = new TreeNode(v);
            node.left = root;
            return node;
        } else {
            Queue<Pair<TreeNode, Integer>> queue = new LinkedList<>();
            queue.add(new Pair<>(root, 1));
            while (!queue.isEmpty()) {
                Pair<TreeNode, Integer> pair = queue.poll();
                TreeNode treeNode = pair.getKey();
                Integer depth = pair.getValue();

                if (depth == d - 1) {
                    TreeNode left = treeNode.left;
                    TreeNode right = treeNode.right;

                    treeNode.left = new TreeNode(v);
                    treeNode.right = new TreeNode(v);
                    treeNode.left.left = left;
                    treeNode.right.right = right;
                } else {
                    if (treeNode.left != null) {
                        queue.add(new Pair<>(treeNode.left, depth + 1));
                    }
                    if (treeNode.right != null) {
                        queue.add(new Pair<>(treeNode.right, depth + 1));
                    }
                }
            }
            return root;
        }
    }