根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/
9 20
/
15 7
https://leetcode-cn.com/explore/learn/card/data-structure-binary-tree/4/conclusion/15/
**思路
我们知道后序遍历的最后一个结点是整个树的根结点,结合中序遍历找到在中序遍历中的根节点就可以分出左右子树,
再通过递归建树。
关键在于在后续遍历中找根节点和通过根节点分左右的参数怎么写出来
我们找到那个根节点也就是划分左右子树的边界,再中序遍历中找出左右子树的长度,再再后序遍历中划分左右子树对于的区间。我们可以知道中序遍历和后续遍历在下标上是不对应的,但是子树的长度都是一样的
所以我们要找好左右子树的长度。**
class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
if(inorder==null||inorder.length==0)
return null;
return BFS(inorder,0,inorder.length-1,postorder,0,postorder.length-1);
}
private TreeNode BFS(int[] inorder, int i, int j, int[] postorder, int k, int l) {
if(i>j) //临界条件
return null;
TreeNode node=new TreeNode(postorder[l]); //后序遍历的最后一个结点是根节点
int border=0; //在中序遍历中找到左右子树的分界点
for(int p=i;p<=j;p++) {
if(inorder[p]==postorder[l]) {
border=p;
break;
}
}
int leftl=border-i; //左子树的长度
int rightl=j-border; //右子树的长度
node.left=BFS(inorder,i,border-1,postorder,k,k+leftl-1);
/*对于中序遍历应该是边界的左边,对于后序遍历应该是从开头长度为左子树长度*/
node.right=BFS(inorder,border+1,j,postorder,l-rightl,l-1);
/*对于中序遍历应该是边界的右边,对于后序遍历应该是从结尾-右子树长度到结尾-1*/
return node;
}
// public static void main(String[] A) {
// int []inorder= {9,3,15,20,7};
// int []postorder= {9,15,7,20,3};
// TreeNode node=new Solution().buildTree(inorder, postorder);
// }
}