JAVA程序设计：在二叉树中增加一行（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    

扫描二维码关注公众号，回复： 9038383 查看本文章

v = 1

d = 3

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

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

思路：想不通这么个辣鸡题才过这么点人？？？

解法很简单，开个临时变量存一下要加的那一层的下边所有的子树即可。。。。我只能秀下我程序的耗时和内存占用了哈哈

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(d==1) {
    		TreeNode myRoot=new TreeNode(v);
    		myRoot.left=root;
    		return myRoot;
    	}
        dfs(root,v,d,1);
        return root;
    }
    
    private void dfs(TreeNode root,int v,int d,int stepth) {
    	if(root==null) return;
    	if(stepth+1==d) {
    		TreeNode tmp=new TreeNode(v);
    		tmp.left=root.left;root.left=tmp;
    		TreeNode tmp1=new TreeNode(v);
    		tmp1.right=root.right;root.right=tmp1;
    		return;
    	}
    	dfs(root.left,v,d,stepth+1);
    	dfs(root.right,v,d,stepth+1);
    }
}