给定一个二叉树,根节点为第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 = 1d = 2输出: 4/ \1 1/ \2 6/ \ / 3 1 5
示例 2:
输入: 二叉树如下所示:4/ 2 / \ 3 1 v = 1d = 3输出: 4/ 2/ \ 1 1/ \ 3 1
注意:
- 输入的深度值 d 的范围是:[1,二叉树最大深度 + 1]。
- 输入的二叉树至少有一个节点。
class Solution {
public:TreeNode* addOneRow(TreeNode* root, int v, int d) {if(!root) return root;if(d == 1){TreeNode *temp = new TreeNode(v);temp->left = root;temp->right = NULL;return temp;}queue<TreeNode *> qu;qu.push(root);int n = 0;while(!qu.empty()){int count = qu.size();++n;while(count--){TreeNode *s = qu.front();qu.pop();if(n == d -1){TreeNode *temp1 = new TreeNode(v);temp1->left = s->left;temp1->right = NULL;s->left = temp1;TreeNode *temp2 = new TreeNode(v);temp2->right = s->right;temp2->left = NULL;s->right = temp2;}if(s->left) qu.push(s->left);if(s->right) qu.push(s->right);}}return root;}
};