leetcode 623. Add One Row to Tree（ 在二叉树中增加一行）

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

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

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

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

示例 1:

Input: 
A binary tree as following:
       4
     /   \
    2     6
   / \   / 
  3   1 5   

v = 1

d = 2

Output: 
       4
      / \
     1   1
    /     \
   2       6
  / \     / 
 3   1   5   

思路：
因为要某一层的节点，所以考虑用BFS，到某一层时，把每个节点的左子树换成TreeNode(v), TreeNode(v)的左子树换成原节点的左子树。同理右子树。

    public TreeNode addOneRow(TreeNode root, int v, int d) {
        if(root == null) {
            return new TreeNode(v);
        };
        
        if(d == 1) {
            TreeNode newRoot = new TreeNode(v);
            newRoot.left = root;
            return newRoot;
        }
        
        int depth = 1;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        while(!queue.isEmpty()) {
            int size = queue.size();
            
            for(int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if(node == null) continue;
                if(d-1 == depth) {
                    TreeNode newL = new TreeNode(v);
                    TreeNode newR = new TreeNode(v);
                    newL.left = node.left;
                    newR.right = node.right;
                    node.left = newL;
                    node.right = newR;
                } else {
                    queue.offer(node.left);
                    queue.offer(node.right);
                }
            }
            
            depth ++;
        }
        
        return root;
    }

上面这个写法当下时间点只beat 33%，需要优化，发现耗时的地方在于null节点也给加进queue了，而null节点又是不需要处理的。另外，在建立new TreeNode时，可以在构造函数中直接指定左子树和右子树。
优化后当下时间点beat 100%

    public TreeNode addOneRow(TreeNode root, int v, int d) {
        if(root == null) {
            return new TreeNode(v);
        };
        
        if(d == 1) {
            TreeNode newRoot = new TreeNode(v);
            newRoot.left = root;
            return newRoot;
        }
        
        int depth = 1;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        while(!queue.isEmpty()) {
            int size = queue.size();
            
            for(int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if(d-1 == depth) {
                    node.left = new TreeNode(v, node.left, null);
                    node.right = new TreeNode(v, null, node.right);
                } else {
                    if(node.left != null) queue.offer(node.left);
                    if(node.right != null) queue.offer(node.right);
                }
            }
            
            depth ++;
        }
        
        return root;
    }