算法小知识-----8.05-----在二叉树中增加一行

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

代替公司出差了两天，居然还漏了一天的更文，实在是遗憾

在二叉树中增加一行

该题出自力扣的623题 —— 在二叉树中增加一行【中等题】

审题

给定一个二叉树的根 root 和两个整数 val 和 depth ，在给定的深度 depth 处添加一个值为 val 的节点行。
注意，根节点 root 位于深度 1 。
加法规则如下:

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

• 题意并不复杂，就是给出一个二叉树，并且在给定的深度层次里面新增一行子树，并且对原来的子树依附在新的子树下

• 结合题意，可以得出两种解答 —— 广度遍历、深度遍历

  • 广度遍历，也就是利用堆栈进行存储每层的子树，并且用当前堆栈的长度判断一层；
    • 在遍历到指定深度，也就是层数时，新增一个子树，使用三个参数的构造函数，对左右子树进行赋值
  • 深度遍历
    • 利用递归，不断对深度的变量进行递减，直到打到指定的深度层数

• 需要注意的是，当深度为1或者0时，意味着插入的是根节点

编码

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> stack = new ArrayDeque<>();
        int de = 1;
        stack.offer(root);
        while (!stack.isEmpty()){
            int size = stack.size();
            if (de == depth - 1){
                for (int i = 0; i < size; i++) {
                    TreeNode pop = stack.poll();
                    pop.left = new TreeNode(val,pop.left,null);
                    pop.right = new TreeNode(val,null,pop.right);
                }
                break;
            }else {
                for (int i = 0; i < size; i++) {
                    TreeNode pop = stack.poll();
                    if (pop.left != null) {
                        stack.offer(pop.left);
                    }
                    if (pop.right != null) {
                        stack.offer(pop.right);
                    }
                }
            }
            de++;
        }
        return root;
    }
}