leetcode 623.在二叉树中增加一行

给定一个二叉树，根节点为第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

注意:

输入的深度值 d 的范围是：[1，二叉树最大深度 + 1]。
输入的二叉树至少有一个节点。
以下是代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode addOneRow(TreeNode root, int v, int d) {
        if(root == null)return root;
        if(d == 1){
            TreeNode node = new TreeNode(v);
            node.left = root;
            return node;
        }
        add(root,v,d,1);
        return root;
    }
    
    public void add(TreeNode root, int v, int d, int depth){
        if(root == null)return;
        depth ++ ;
        if(depth == d){
            TreeNode lnode = new TreeNode(v);
            lnode.left = root.left;
            root.left = lnode;
            TreeNode rnode = new TreeNode(v);
            rnode.right = root.right;
            root.right = rnode;
            return;
        }
        add(root.left,v,d,depth);
        add(root.right,v,d,depth);
    }
    
}

（击败了100% 有点小激动，专门记录一下。。）