1. 题目

给定一个二叉树，根节点为第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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

2.1 BFS

用队列按层遍历，到达指定的层，在层间加入新的节点，再将新节点与上下层连接起来

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;//存储下层节点leftr= tp->right;//存储下层节点righttp->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;}
};

2.2 DFS

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);}
};