文章目錄

• • 1. 題目
  • 2. 解題
  • • 2.1 BFS
    • 2.2 DFS

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

總結

以上是生活随笔為你收集整理的LeetCode 623. 在二叉树中增加一行（BFS/DFS）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。