1. 前序与中序遍历序列构造二叉树
描述:
给定一棵树的前序遍历 preorder
与中序遍历 inorder
。请构造二叉树并返回其根节点
思路:
根据前序遍历找根
(1)定义成员变量preIndex来遍历前序遍历数组
定义一个辅助方法buildTreeChild来构建树,参数为(前序遍历数组,中序遍历数组,中序遍历开始下标,中序遍历结束下标),返回值为树的根,若中序遍历开始下标>中序遍历结束下标,代表没有数组区间(没有左子树或者右子树),返回null
(2)从前序遍历数组中获取根节点,new一个节点参数为前序遍历数组的前序索引
(3)有根节点后要在中序遍历数组中找到根的下标,
需要定义一个辅助方法findIndex,参数为(中序遍历数组,开始下标,结束下标,寻找的值)
方法体为:i=开始下标,在i<=结束下标时循环判断中序遍历数组i位置数否等于寻找的值,等于返回i,否则返回-1,每次循环i++
(4)buildTreeChild调用辅助方法findIndex,传入寻找的值就是前序遍历数组的前序索引,
返回值就是中序遍历数组中根节点的下标赋给index保存,找到一个后就定了一个根,前序 索引++
(5)分别递归左右子树,注意参数左子树中序遍历范围是开始下标~index-1,右子树中序遍历 范围是开index+1~结束下标
(6)主函数中,传入的前序遍历和中序遍历的数组只要有一个为空,直接返回null
否则返回辅助方法buildTreeChild的方一致即可
代码:
class Solution {
public int preIndex = 0;
public TreeNode buildTreeChild(int[] preorder, int[] inorder, int inbegin, int inend){
if(inbegin > inend){
return null;
}
TreeNode root = new TreeNode(preorder[preIndex]);
int index = findIndex(inorder, inbegin, inend, preorder[preIndex]);
preIndex++;
root.left = buildTreeChild(preorder, inorder, inbegin, index - 1);
root.right = buildTreeChild(preorder, inorder, index + 1, inend);
return root;
}
public int findIndex(int[] inorder, int inbegin, int inend, int key){
for(int i = inbegin;i <= inend;i++){
if(inorder[i] == key){
return i;
}
}
return -1;
}
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder == null || inorder == null){
return null;
}
return buildTreeChild(preorder,inorder,0,inorder.length - 1);
}
}
2. 中序与后序遍历序列构造二叉树
描述:
根据一棵树的中序遍历与后序遍历构造二叉树。
思路:
和前中的区别:
根是从后往前看的,树是先构建右子树再构建左子树
所以遍历开始的下标改为preIndex = postorder.length - 1;
每次找到一个中序遍历数组根节点位置后preIndex--
代码:
class Solution {
public int preIndex = 0;
public TreeNode buildTreeChild(int[] postorder, int[] inorder, int inbegin, int inend){
if(inbegin > inend){
return null;
}
TreeNode root = new TreeNode(postorder[preIndex]);
int index = findIndex(inorder, inbegin, inend, postorder[preIndex]);
preIndex--;
root.right = buildTreeChild(postorder, inorder, index + 1, inend);
root.left = buildTreeChild(postorder, inorder, inbegin, index - 1);
return root;
}
public int findIndex(int[] inorder, int inbegin, int inend, int key){
for(int i = inbegin;i <= inend;i++){
if(inorder[i] == key){
return i;
}
}
return -1;
}
public TreeNode buildTree(int[] inorder, int[] postorder ) {
if(postorder == null || inorder == null){
return null;
}
preIndex = postorder.length - 1;
return buildTreeChild(postorder,inorder,0,inorder.length - 1);
}
}