在二叉树中增加一行 · 8 月更文挑战

55 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第22天,点击查看活动详情

一、题目描述:

给定一个二叉树的根 root 和两个整数 val 和 depth ,在给定的深度 depth 处添加一个值为 val 的节点行。

注意,根节点 root 位于深度 1 。

加法规则如下:

给定整数 depth,对于深度为 depth - 1 的每个非空树节点 cur ,创建两个值为 val 的树节点作为 cur 的左子树根和右子树根。 cur 原来的左子树应该是新的左子树根的左子树。 cur 原来的右子树应该是新的右子树根的右子树。 如果 depth == 1 意味着 depth - 1 根本没有深度,那么创建一个树节点,值 val 作为整个原始树的新根,而原始树就是新根的左子树。

示例 1:

img 输入: root = [4,2,6,3,1,5], val = 1, depth = 2 输出: [4,1,1,2,null,null,6,3,1,5] 示例 2:

img

输入: 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

来源:力扣(LeetCode) 链接:leetcode.cn/problems/ad… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二、思路分析:

利用广度优先搜索找到depth - 1行的所有结点,然后在这行所有节点与其子节点之间插入指定的新节点就可以了。

注意两点。 1.如果(depth - 1)层的节点没有子节点,需要两个子节点都补充上 2.在root层添加节点的时候,原root自动成为新root的左节点(给leetcode意见,该用例应该删除)

三、AC 代码:

​
class Solution {
    public TreeNode addOneRow(TreeNode root, int val, int depth) {
        if (depth == 1) {
            TreeNode node = new TreeNode(val);
            node.left = root;
            return node;
        }
​
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);
        int curDepth = 1;
​
        while (depth > curDepth + 1) {
            List<TreeNode> list = new ArrayList<>();
            while (!queue.isEmpty()) {
                TreeNode node = queue.poll();
                if (node.left != null) {
                    list.add(node.left);
                }
                if (node.right != null) {
                    list.add(node.right);
                }
            }
            curDepth++;
            queue.addAll(list);
        }
​
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            TreeNode left = new TreeNode(val);
            TreeNode right = new TreeNode(val);
            left.left = node.left;
            right.right = node.right;
            node.left = left;
            node.right = right;
        }
        
        return root;
    }
}
​

四、总结:

image.png

掘友们,解题不易,留下个赞或评论再走吧!谢啦~ 💐

希望对你有帮助

期待下次再见~

🌇 点赞 👍 收藏 ⭐留言 📝 一键三连 ~关注,从你我做起!