Leetcode_623_在二叉树的中增加一行_数据结构

九幽孤翎

于 2022-08-05 17:31:25 发布

阅读量293

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构 daily_algorithm 文章标签： leetcode 数据结构深度优先

本文链接：https://blog.youkuaiyun.com/HDUCheater/article/details/126182333

daily_algorithm 同时被 2 个专栏收录

230 篇文章

订阅专栏

数据结构

52 篇文章

订阅专栏

这个博客介绍了一个Java方法，用于在一棵给定的二叉树的指定深度处插入一个新行，每个新节点的值都是固定的。通过深度优先搜索实现，将原有子树挂载到新节点的两侧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

该说不说的，我觉得写的一气呵成挺完美

class Solution {
    public TreeNode addOneRow(TreeNode root, int val, int depth) {
        if(depth == 1) {
            TreeNode node = new TreeNode(val);
            node.left = root;
            return node;
        }
        dfs(root, 2, depth, val);
        return root;
    }

    void dfs(TreeNode node, int nowDepth, int targetDepth, int val) {
        if(node == null) {
            return;
        }
        if(nowDepth == targetDepth) {
            process(node, val);
        }
        dfs(node.left, nowDepth + 1, targetDepth, val);
        dfs(node.right, nowDepth + 1, targetDepth, val);
    }

    /**
     * 处理cur，为其添加两个值为val的子节点，并把原来的子树挂载在两端
     */
    void process(TreeNode node, int val) {
        TreeNode left = node.left;
        TreeNode right = node.right;
        node.left = new TreeNode(val);
        node.right = new TreeNode(val);
        node.left.left = left;
        node.right.right = right;
    }
}