leetcode623_在二叉树中增加一行

给定一个二叉树,根节点为第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]。
输入的二叉树至少有一个节点。

 

思路:

利用bfs的思想 利用队列保存第d-1行的节点   再进行插入

 if (d == 1) {
	            TreeNode newroot = new TreeNode(v);
	            newroot.left = root;
	            return newroot;
	        }
	        LinkedList<TreeNode> queue = new LinkedList<>();
	        queue.add(root);
	        for (int i = 0; i < d-2; i++) {
	            int size = queue.size();
	            for (int j = 0; j < size; j++) {
	                TreeNode t = queue.poll();
	                if (t.left != null) queue.add(t.left);
	                if (t.right != null) queue.add(t.right);
	            }
	        }
	        //此时队列中保存d-1行的节点
	        while (!queue.isEmpty()) {
	            TreeNode t = queue.poll();
	            //处理左孩子
	            TreeNode tmp = t.left;
	            t.left = new TreeNode(v);    //将新的节点v加入
	            t.left.left = tmp;
	            
	            //处理右孩子
	            tmp = t.right;
	            t.right = new TreeNode(v);   //将新的节点v加入
	            t.right.right = tmp;
	        }
	        return root;

 

全部评论

相关推荐

点赞 评论 收藏
分享
威猛的小饼干正在背八股:挂到根本不想整理
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务