JAVA程序设计：完全二叉树插入器（LeetCode：919）

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

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

CBTInserter(TreeNode root) 使用头结点为 root 的给定树初始化该数据结构；
CBTInserter.insert(int v) 将 TreeNode 插入到存在值为 node.val = v  的树中以使其保持完全二叉树的状态，并返回插入的 TreeNode 的父结点的值；
CBTInserter.get_root() 将返回树的头结点。
 

示例 1：

输入：inputs = ["CBTInserter","insert","get_root"], inputs = [[[1]],[2],[]]
输出：[null,1,[1,2]]
示例 2：

输入：inputs = ["CBTInserter","insert","insert","get_root"], inputs = [[[1,2,3,4,5,6]],[7],[8],[]]
输出：[null,3,4,[1,2,3,4,5,6,7,8]]
 

提示：

最初给定的树是完全二叉树，且包含 1 到 1000 个结点。
每个测试用例最多调用 CBTInserter.insert  操作 10000 次。
给定结点或插入结点的每个值都在 0 到 5000 之间。

思路：我们标记最下边一层需要放多少个和已经放了多少个，若放满了就再加一层，对于具体放在最底层的那个位置，我们考虑类似于二分查找的模式，例如：最底层放满需要8个元素，目前已经放了5个了，现在需要放第六个元素。则第一次应该走到根节点的右子树，因为左子树的最底层最多也只能放4个元素，则我们将右子树的树根当做根节点，而因为左子树已经包含了4个元素，因此在当前子树中最底层只用将待插元素放在第6-4个位置上即可，其实你可以发现问题已经归结为子问题了，进一下向下也是一样的。

class CBTInserter {

	int sum;
	int size=0,nowNum;
	TreeNode root,start;
	
    public CBTInserter(TreeNode root) {
    	
    	sum=0;
    	start=root;
    	this.root=root;
    	
    	dfs(this.root);
    	
    	int len=0,sm=0;
    	while(sm+(1<<len)<=sum) {
    		sm+=1<<len;
    		len++;
    	}
    	
    	root=start;
    	nowNum=sum-sm+1;	
    	size=1<<len;
    }
    
    private void dfs(TreeNode root) {
    	if(root==null) return;
    	sum++;
    	dfs(root.left);
    	dfs(root.right);
    }
    
    public int insert(int v) {
    	
    	root=start;
    	int tmp1=nowNum,tmp2=size;
    	
    	while(root!=null) {
    		if(tmp1>tmp2/2) {
    			tmp1-=tmp2/2;
    			if(tmp2>2)
    				root=root.right;
    			else {
    				root.right=new TreeNode(v);
    				break;
    			}
    		}
    		else {
    			if(tmp2>2)
    				root=root.left;
    			else {
    				root.left=new TreeNode(v);
    				break;
    			}
    		}
    		tmp2/=2;
    	}
    	
    	if(nowNum==size) {
    		size*=2;
    		nowNum=1;
    	}
    	else
    		nowNum++;
    	
    	return root.val;
    }
    
    public TreeNode get_root() {
    	return start;
    }
}