java自习室 day21

**leetcode 623题 在二叉树中增加一行 **
给定一个二叉树,根节点为第1层,深度为 1。在其第 d 层追加一行值为 v 的节点。

添加规则:给定一个深度值 d (正整数),针对深度为 d-1 层的每一非空节点 N,为 N 创建两个值为 v 的左子树和右子树。

将 N 原先的左子树,连接为新节点 v 的左子树;将 N 原先的右子树,连接为新节点 v 的右子树。

如果 d 的值为 1,深度 d - 1 不存在,则创建一个新的根节点 v,原先的整棵树将作为 v 的左子树。

示例 1:

输入:
二叉树如下所示:
4
/
2 6
/ \ /
3 1 5

v = 1

d = 2

输出:
4
/
1 1
/
2 6
/ \ /
3 1 5

考察二叉树的层序遍历,遍历到目标深度的上一层时,添加目标节点,代码如下:

class Solution {
    public TreeNode addOneRow(TreeNode root,int v,int d){
        if (d == 1){
            TreeNode newRoot = new TreeNode(v);
            newRoot.left = root;
            return newRoot;
        }

        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        int depth = 0;
        while(!queue.isEmpty()){
            depth++;
            if(depth == d - 1){
                while(!queue.isEmpty()){
                    TreeNode node = queue.remove();
                    TreeNode temp = node.left;
                    node.left = new TreeNode(v);
                    node.left.left = temp;
                    temp = node.right;
                    node.right = new TreeNode(v);
                    node.right.right = temp;
                }
                break;
            }
            int n = queue.size();
            while(n>0){
                TreeNode cur = queue.remove();
                if(cur.left != null) {
                    queue.add(cur.left);
                }
                if(cur.right != null) {
                    queue.add(cur.right);
                }
                n--;
            }   
        }
    return root;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值