**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
考察二叉树的层序遍历,遍历到目标深度的上一层时,添加目标节点,代码如下:
class Solution {
public TreeNode addOneRow(TreeNode root,int v,int d){
if (d == 1){
TreeNode newRoot = new TreeNode(v);
newRoot.left = root;
return newRoot;
}
Queue<TreeNode> queue = new ArrayDeque<>();
queue.add(root);
int depth = 0;
while(!queue.isEmpty()){
depth++;
if(depth == d - 1){
while(!queue.isEmpty()){
TreeNode node = queue.remove();
TreeNode temp = node.left;
node.left = new TreeNode(v);
node.left.left = temp;
temp = node.right;
node.right = new TreeNode(v);
node.right.right = temp;
}
break;
}
int n = queue.size();
while(n>0){
TreeNode cur = queue.remove();
if(cur.left != null) {
queue.add(cur.left);
}
if(cur.right != null) {
queue.add(cur.right);
}
n--;
}
}
return root;
}
}