当前位置：网站首页>623. Add a row to a binary tree: Simple binary tree traversal problems

623. Add a row to a binary tree: Simple binary tree traversal problems

2022-08-05 11:38:00 【Mitsuha Miyamizu's writing diary】

题目描述

这是 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 The entire layer expand,At the same time record the current depth,当到达第 depth - 1 层,Is to add some operation.

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 Also can be carried out to solve,在 DFS In the process of recording the current depth.

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