利用先序、后序、中序两两组合重构二叉树(以Java为例)
二叉树的相关问题中利用先序、后序、中序两两结合来重构二叉树是比较常见的一类问题,对于这一类的问题我们可以利用递归的思想来解决。我将结合下面的例子来进行描述。
先序:1 2 4 5 8 9 3 6 7
中序:4 2 8 5 9 1 6 3 7
后序:4 8 9 5 2 6 7 3 1
说明:
pre : 先序 prestart : 先序开始位置 preend : 先序结束位置
in : 中序 instart : 中序开始位置 inend : 中序结束位置
pos : 后序 posstart : 后序开始位置 posend : 后序结束位置
index : 根节点在中序中的位置
一、利用先序和中序重构二叉树
已知:
先序:1 2 4 5 8 9 3 6 7
中序:4 2 8 5 9 1 6 3 7
先序中的第一个节点 1 即为整个二叉树的根节点,在中序中找到节点 1 的位置,由中序遍历的特点,可知 1 左边为根节点的左子树, 1 右边为根节点的右子数。以此规律遍历二叉树根节点的左子树和右子数,即可重构二叉树。
整个二叉树:
pre : prestart = 0,preend = 8
in : instart = 0,inend = 8
index = 5
左子树:
pre : prestart = prestart+1,preend = prestart+1+index-instart-1
in : instart = instart,inend = index-1
右子数:
pre : prestart=prestart+index-instart+1,preend = preend
in : instart = index+1, inend = inend
public Node preInToTree(int[] pre,int[] in){
if(pre==null||in==null){
return null;
}
HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
for(int i=0;i<pre.length;i++){
map.put(in[i],i);
}
return preIn(pre,0,pre.length-1,in,0,in.length-1,map);
}
public Node preIn(int[] pre,int prestart,int preend,int[] in,int instart,int inend,HashMap<Integer,Integer> map){
if(prestart>preend)
return null;
Node head = new Node(pre[prestart]);
int index = map.get(pre[prestart]);
head.left = preIn(pre,prestart+1,prestart+1+index-instart-1,in,instart,index-1,map);
head.right = preIn(pre,prestart+index-instart+1,preend,in,index+1,inend,map);
}
二、利用后序和中序重构二叉树
同理可得:
整个二叉树:
pre : prestart = 0,preend = 8
in : instart = 0,inend = 8
index = 5
左子树:
pos : posstart = posstart , posend = posstart+index-instart-1
in : instart = instart,inend = index-1
右子数:
pos : posstart = posstart+index-instart , posend = posend -1
in : instart = index+1, inend = inend
代码与之前类似,这里不再赘述
三、先序和后序重构二叉树
利用先序和后序重构二叉树,一般情况下无法重构出唯一的二叉树,结果往往有多个,意义不大,因此这里不作讨论。