当前位置:网站首页>623. 在二叉树中增加一行 : 简单二叉树遍历运用题
623. 在二叉树中增加一行 : 简单二叉树遍历运用题
2022-08-05 11:31:00 【宫水三叶的刷题日记】
题目描述
这是 LeetCode 上的 623. 在二叉树中增加一行 ,难度为 中等。
Tag : 「二叉树」、「BFS」、「DFS」
给定一个二叉树的根 root
和两个整数 val
和 depth
,在给定的深度 depth
处添加一个值为 val
的节点行。
注意,根节点 root
位于深度 。
加法规则如下:
给定整数 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]
提示:
节点数在 范围内 树的深度在 范围内
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
};
时间复杂度: 空间复杂度:
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 = 0
function 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)
}
}
时间复杂度: 空间复杂度:
最后
这是我们「刷穿 LeetCode」系列文章的第 No.623
篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,部分是有锁题,我们将先把所有不带锁的题目刷完。
在这个系列文章里面,除了讲解解题思路以外,还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。
为了方便各位同学能够电脑上进行调试和提交代码,我建立了相关的仓库:https://github.com/SharingSource/LogicStack-LeetCode 。
在仓库地址里,你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。
更多更全更热门的「笔试/面试」相关资料可访问排版精美的 合集新基地
边栏推荐
猜你喜欢
365 days challenge LeetCode1000 questions - Day 050 add a row to the binary tree binary tree
2022杭电多校联赛第六场 题解
Flink Yarn Per Job - 启动TM,向RM注册,RM分配solt
结合“xPlus”探讨软件架构的创新与变革
DocuWare平台——文档管理的内容服务和工作流自动化的平台详细介绍(下)
解决2022Visual Studio中scanf返回值被忽略问题
TiDB 6.0 Placement Rules In SQL 使用实践
力扣330 按要求补齐数组(贪心)
苹果Meta都在冲的Pancake技术,中国VR团队YVR竟抢先交出产品答卷
【硬件架构的艺术】学习笔记(3)处理多个时钟
随机推荐
互联网行业凛冬之至,BATM的程序员是如何应对中年危机的?
【AGC】增长服务1-远程配置示例
再获殊荣 | 赛宁网安入选2022年度“培育独角兽”企业榜单
How OpenHarmony Query Device Type
trie树模板
Android 开发用 Kotlin 编程语言一 基本数据类型
gradle尚硅谷笔记
shell编程流程控制练习
PPOCR 检测器配置文件参数详解
智能算力的枢纽如何构建?中国云都的淮海智算中心打了个样
Linux: Remember to install MySQL8 on CentOS7 (blog collection)
微信小程序标题栏封装
平安萌娃卡保险怎么样?让父母读懂几个识别产品的方法
Hands-on Deep Learning_GoogLeNet / Inceptionv1v2v3v4
使用Windbg过程中两个使用细节分享
深度学习(四)分析问题与调参 理论部分
.NET in-depth analysis of the LINQ framework (6: LINQ execution expressions)
Machine Learning - Logistic Regression
手把手教你定位线上MySQL慢查询问题,包教包会
【加密解密】明文加密解密-已实现【已应用】