leetcode 919. 完全二叉树插入器-题解

919. 完全二叉树插入器

题目链接

题目描述

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

  设计一种算法，将一个新节点插入到一个完整的二叉树中，并在插入后保持其完整。

  实现 CBTInserter 类:

  • CBTInserter(TreeNode root) 使用头节点为 root 的给定树初始化该数据结构；
  • CBTInserter.insert(int v) 向树中插入一个值为 Node.val == val的新节点 TreeNode。使树保持完全二叉树的状态，并返回插入节点 TreeNode 的父节点的值；
  • 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[] t = {root};
        return insertNode(t,node);
    }
    
    public TreeNode get_root() {
        return root;
    }

    /**
     * 当保存非叶子节点的数组存在叶子节点或存在不是满二叉树的节点时
     * 就找到了要插入节点的位置
     * @param troot 保存第n层非叶子节点的数组
     * @param node 要插入的值
     */
    private int insertNode(TreeNode[] troot,TreeNode node){
        //循环当前层非叶子节点
        for (TreeNode treeNode : troot) {
            //判断当前节点左子节点是否为空
            if (treeNode.left == null){
                treeNode.left = node;
                return treeNode.val;
            }
            //判断当前节点右子节点是否为空
            if (treeNode.right == null){
                treeNode.right = node;
                return treeNode.val;
            }
        }
        //循环结束表示当前的节点层没有插入的位置
        //准备开始递归下一层节点
        //保存下一层的节点数组,数组大小是当前节点层的两倍
        TreeNode[] t = new TreeNode[troot.length << 1];

        //将下一层的节点加入节点数组
        int count = 0;
        for (TreeNode treeNode : troot) {
            t[count++] = treeNode.left;
            t[count++] = treeNode.right;
        }

       return insertNode(t,node);
    }
}

/**
 * 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 “Offer II 043. 往完全二叉树添加节点” 题类似