leetcode 剑指 Offer II 043. 往完全二叉树添加节点-题解

剑指 Offer II 043. 往完全二叉树添加节点

题目链接

题目描述

• 完全二叉树是每一层（除最后一层外）都是完全填充（即，节点数达到最大，第 n 层有 2n-1 个节点）的，并且所有的节点都尽可能地集中在左侧。

  设计一个用完全二叉树初始化的数据结构 CBTInserter，它支持以下几种操作：

  • CBTInserter(TreeNode root) 使用根节点为 root 的给定树初始化该数据结构；
  • CBTInserter.insert(int v) 向树中插入一个新节点，节点类型为 TreeNode，值为 v 。使树保持完全二叉树的状态，并返回插入的新节点的父节点的值；
  • CBTInserter.get_root() 将返回树的根节点。

算法分析

• 使用广度优先遍历查找要插入的位置
• 将树分为n层,根节点为第一层,将根节点放入保存当前层的节点数组
• 遍历当前层的数组,判断是否有节点的子节点是空的,如果有,则将节点插入到它的子节点位置,并返回它的值
• 如果没有节点的子节点是空的,则将当前层的所有子节点放到下一层节点的数组,并开始遍历下一层的节点数组
• 直到节点插入成功

代码实现

/**
 * Definition for a binary tree node.
 * 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;
 *     }
 * }
 */
class CBTInserter {
        TreeNode root;
    public CBTInserter(TreeNode root) {
        this.root  = root;
    }
    
    public int insert(int val) {
        //创建节点
        TreeNode node = new TreeNode(val);
        //创建只包含根节点节点数组
        TreeNode[] troot = {root};

        while(true){
            //保存下一层的节点数组,数组大小是当前节点层的两倍
            TreeNode[] t = new TreeNode[troot.length << 1];
            int count = 0;

            //循环当前层非叶子节点
            for (TreeNode treeNode : troot) {
                //判断当前节点左子节点是否为空
                if (treeNode.left == null){
                    treeNode.left = node;
                    return treeNode.val;
                }
                //判断当前节点右子节点是否为空
                if (treeNode.right == null){
                    treeNode.right = node;
                    return treeNode.val;
                }
                t[count++] = treeNode.left;
                t[count++] = treeNode.right;
            }
            //开始下一层的遍历
            troot = t;
        }
    }
    
    public TreeNode get_root() {
        return root;
    }
    
}

/**
 * Your CBTInserter object will be instantiated and called as such:
 * CBTInserter obj = new CBTInserter(root);
 * int param_1 = obj.insert(val);
 * TreeNode param_2 = obj.get_root();
 */

涉及知识点

• 树
• 完全二叉树
• 递归改循环
• 数组

该题与leetcode “919. 完全二叉树插入器” 题类似