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

最新推荐文章于 2025-08-03 18:35:24 发布

原创最新推荐文章于 2025-08-03 18:35:24 发布 · 317 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #数据结构 #深度优先

daily_algorithm 同时被 2 个专栏收录

230 篇文章

订阅专栏

数据结构

52 篇文章

订阅专栏

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

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

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;
    }
}