携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第5天,点击查看活动详情
前言
今天的题目为中等,但是题目整体比较简单,通过简单的BFS深度优先遍历就可以得出答案,也没有什么绕圈子的地方。
每日一题
今天的题目是 623. 在二叉树中增加一行,难度为中等
-
给定一个二叉树的根 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, 104] 范围内
- 树的深度在 [1, 104]范围内
- -100 <= Node.val <= 100
- -105 <= val <= 105
- 1 <= depth <= the depth of tree + 1
题解
BFS
深度优先遍历,按照题目的意思,只要记录一个深度,当当前的深度达到题目的要求,就将当前节点的左右孩子添加到两个新的节点的左右孩子中,新节点的值题目也给出了,简单的深度优先遍历就可以解决
深度优先遍历一般就是通过递归来实现,当depth大于2的时候,我们就将递归到当前节点的左右孩子节点,并且因为向下了一层,那么depth就需要减一,一直到极限情况,也就是depth刚好为2的时候,那么就说明这一层的下一层需要插入新节点,那么就把新节点作为当前的下一层的左孩子和右孩子,并且当前节点原来的左孩子和右孩子,变成两个新节点的左孩子和右孩子。
- 如果
depth == 1意味着depth - 1根本没有深度,那么创建一个树节点,值val作为整个原始树的新根,而原始树就是新根的左子树。
通过题目中的这句话我们就能够看出来特殊情况,就是当depth为1的时候,那么就直接新建一颗树,以val作为根节点的值,原来的树作为这个树的左孩子生成一个新树,这就是特殊情况的处理方式
/**
* Definition for a binary tree node.
* class TreeNode {
* val: number
* left: TreeNode | null
* right: TreeNode | null
* constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
* }
*/
function addOneRow(root: TreeNode | null, val: number, depth: number): TreeNode | null {
if (!root) {
return null;
}
if (depth === 1) {
return new TreeNode(val, root, null);
}
if (depth === 2) {
root.left = new TreeNode(val, root.left, null);
root.right = new TreeNode(val, null, root.right);
} else {
root.left = addOneRow(root.left, val, depth - 1);
root.right = addOneRow(root.right, val, depth - 1);
}
return root;
};