题目形容
这是 LeetCode 上的 623. 在二叉树中减少一行 ，难度为 中等。
Tag : 「二叉树」、「BFS」、「DFS」
给定一个二叉树的根 root 和两个整数 val 和 depth，在给定的深度 depth 处增加一个值为 val 的节点行。
留神，根节点 root 位于深度 $1$ 。
加法规定如下:
- 给定整数
depth，对于深度为depth - 1的每个非空树节点cur，创立两个值为val的树节点作为cur的左子树根和右子树根。 cur原来的左子树应该是新的左子树根的左子树。cur原来的右子树应该是新的右子树根的右子树。- 如果
depth == 1意味着depth - 1基本没有深度，那么创立一个树节点，值val作为整个原始树的新根，而原始树就是新根的左子树。
示例 1:
输出: root = [4,2,6,3,1,5], val = 1, depth = 2输入: [4,1,1,2,null,null,6,3,1,5]示例 2:
输出: root = [4,2,null,3,1], val = 1, depth = 3输入: [4,2,null,1,1,3,null,null,1]提醒:
- 节点数在 $[1, 10^4]$ 范畴内
- 树的深度在 $[1, 10^4]$范畴内
- $-100 <= Node.val <= 100$
- $-10^5 <= val <= 10^5$
- $1 <= depth <= the depth of tree + 1$
BFS
依据 BFS 来做，每次 BFS 将整一层进行拓展，同时记录以后深度，当达到第 depth - 1 层，则进行加点操作。
Java 代码：
class Solution { public TreeNode addOneRow(TreeNode root, int val, int depth) { if (depth == 1) { TreeNode ans = new TreeNode(val); ans.left = root; return ans; } Deque<TreeNode> d = new ArrayDeque<>(); d.addLast(root); int cur = 1; while (!d.isEmpty()) { int sz = d.size(); while (sz-- > 0) { TreeNode t = d.pollFirst(); if (cur == depth - 1) { TreeNode a = new TreeNode(val), b = new TreeNode(val); a.left = t.left; b.right = t.right; t.left = a; t.right = b; } else { if (t.left != null) d.addLast(t.left); if (t.right != null) d.addLast(t.right); } } cur++; } return root; }}TypeScript 代码：
function addOneRow(root: TreeNode | null, val: number, depth: number): TreeNode | null { if (depth == 1) { const ans = new TreeNode(val) ans.left = root return ans } const stk = new Array<TreeNode>() let he = 0, ta = 0, cur = 1 stk[ta++] = root while (he < ta) { let sz = ta - he while (sz-- > 0) { const t = stk[he++] if (cur == depth - 1) { const a = new TreeNode(val), b = new TreeNode(val) a.left = t.left; b.right = t.right t.left = a; t.right = b } else { if (t.left != null) stk[ta++] = t.left if (t.right != null) stk[ta++] = t.right } } cur++ } return root};- 工夫复杂度：$O(n)$
- 空间复杂度：$O(n)$
DFS
同理，应用 DFS 也可进行求解，在 DFS 过程中记录以后深度。
Java 代码：
class Solution { int d, v; public TreeNode addOneRow(TreeNode root, int val, int depth) { d = depth; v = val; if (d == 1) { TreeNode ans = new TreeNode(val); ans.left = root; return ans; } dfs(root, 1); return root; } void dfs(TreeNode root, int cur) { if (root == null) return ; if (cur == d - 1) { TreeNode a = new TreeNode(v), b = new TreeNode(v); a.left = root.left; b.right = root.right; root.left = a; root.right = b; } else { dfs(root.left, cur + 1); dfs(root.right, cur + 1); } }}TypeScript 代码：
let d = 0, v = 0function addOneRow(root: TreeNode | null, val: number, depth: number): TreeNode | null { d = depth; v = val if (d == 1) { const ans = new TreeNode(v) ans.left = root return ans } dfs(root, 1) return root};function dfs(root: TreeNode | null, cur: number): void { if (root == null) return if (cur == d - 1) { const a = new TreeNode(v), b = new TreeNode(v) a.left = root.left; b.right = root.right root.left = a; root.right = b } else { dfs(root.left, cur + 1) dfs(root.right, cur + 1) }}- 工夫复杂度：$O(n)$
- 空间复杂度：$O(n)$
最初
这是咱们「刷穿 LeetCode」系列文章的第 No.623 篇，系列开始于 2021/01/01，截止于起始日 LeetCode 上共有 1916 道题目，局部是有锁题，咱们将先把所有不带锁的题目刷完。
在这个系列文章外面，除了解说解题思路以外，还会尽可能给出最为简洁的代码。如果波及通解还会相应的代码模板。
为了不便各位同学可能电脑上进行调试和提交代码，我建设了相干的仓库：https://github.com/SharingSou... 。
在仓库地址里，你能够看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其余优选题解。
更多更全更热门的「口试/面试」相干材料可拜访排版精美的 合集新基地
本文由mdnice多平台公布