剑指offer43:往完全二叉树添加节点

题目:
完全二叉树是每一层(除最后一层外)都是完全填充(即,节点数达到最大,第 n 层有 2^n-1 个节点)的,并且所有的节点都尽可能地集中在左侧。
设计一个用完全二叉树初始化的数据结构 CBTInserter,它支持以下几种操作:
CBTInserter(TreeNode root) 使用根节点为 root 的给定树初始化该数据结构;
CBTInserter.insert(int v) 向树中插入一个新节点,节点类型为 TreeNode,值为 v 。使树保持完全二叉树的状态,并返回插入的新节点的父节点的值;
CBTInserter.get_root() 将返回树的根节点
分析:
在这里插入图片描述
如上图,完全二叉树插入节点7,8,9,观察到插入顺序满足广度优先遍历,数据结构可以采用队列,从根节点开始广度优先遍历,先入队根节点,当一个节点既有左节点又有右节点时,出队该节点,并入队该节点的左节点和该节点的右节点,根据这种规律以此类推,例如a,1出队,2,3入队,2有左右节点,所以2出队,4,5入队,因此队列中存在5,4,3节点,节点6还未添加到队列中。
当插入节点7时,节点3正好位于队头,3已经有左节点了,因此新的节点插入它右子节点的位置,插入节点7后,3节点左节点和右节点就都存在了,因此出队3,入队6,7,后面的插入步骤同理。
根据广度优先搜索的顺序找出二叉树中所有既有左子节点又有右子节点的节点,因此时间复杂度O(n)。空间复杂度也为O(n)。
代码:

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode() {}
    TreeNode(int val) { this.val = val; }
     TreeNode(int val, TreeNode left, TreeNode right) {
         this.val = val;
         this.left = left;
         this.right = right;
    }
  }
import java.util.LinkedList;
import java.util.Queue;

public class CBTInserter {
    private Queue<TreeNode> queue;
    private TreeNode root;
    public CBTInserter(TreeNode root) {
        this.root = root;
        queue = new LinkedList<>();
        queue.offer(root);
         while (queue.peek().left !=null && queue.peek().right !=null){
             TreeNode node = queue.poll();
             queue.offer(node.left);
             queue.offer(node.right);
         }
    }

    public int insert(int v) {
        TreeNode parent = queue.peek();
        TreeNode node = new TreeNode(v);
        if (parent.left == null){
                parent.left = node;
        }else {
            parent.right = node;
            queue.poll();
            queue.offer(parent.left);
            queue.offer(parent.right);
        }
        return parent.val;
    }

    public TreeNode get_root() {
        return this.root;
    }
}

在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

龙崎流河

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值