给定一个二叉树,根节点为第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]。
输入的二叉树至少有一个节点。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/add-one-row-to-tree 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
用队列按层遍历,到达指定的层,在层间加入新的节点,再将新节点与上下层连接起来
class Solution {
public:
TreeNode* addOneRow(TreeNode* root, int v, int d) {
if(d == 1)
{
TreeNode *newRoot = new TreeNode(v);
newRoot->left = root;
return newRoot;
}
int deep = 1, n;
queue<TreeNode*> q;
TreeNode *tp, *l, *r;
q.push(root);
while(!q.empty() && deep < d)
{
n = q.size();
while(n--)
{
tp = q.front();
q.pop();
if(deep == d-1) //到达指定的层了
{
l = tp->left;//存储下层节点left
r= tp->right;//存储下层节点right
tp->left = new TreeNode(v);//新节点
tp->right = new TreeNode(v);
tp->left->left = l;//新节点与下层连接上
tp->right->right = r;
}
if(tp->left)
q.push(tp->left);
if(tp->right)
q.push(tp->right);
}
deep++;
}
return root;
}
};
class Solution {
public:
TreeNode* addOneRow(TreeNode* root, int v, int d) {
if(d == 1)
{
TreeNode *newRoot = new TreeNode(v);
newRoot->left = root;
return newRoot;
}
dfs(root,1,d,v);
return root;
}
void dfs(TreeNode* root, int deep, int &d, int &v)
{
if(!root || deep == d)
return;
if(deep == d-1)
{
TreeNode* l = root->left;
TreeNode* r = root->right;
root->left = new TreeNode(v);
root->right = new TreeNode(v);
root->left->left = l;
root->right->right = r;
}
dfs(root->left,deep+1,d,v);
dfs(root->right,deep+1,d,v);
}
};