题目描述:
给出一棵树的中序遍历和后序遍历,请构造这颗二叉树。
给定的是中序遍历和后序遍历的数组,输出是树。
题目分析:
对于一颗二叉树的构建,现根据两个数组找到树的头结点,中序遍历顺序是:左中右,先遍历左子树再遍历中间节点最后遍历右子树,对于后序遍历顺序是:先遍历左子树,再遍历右子树,最后遍历中间的节点。后序数组的最后一个节点为树的根节点,根据根节点在中序遍历的位置index,index左边的节点为该节点的左树,右边节点为该节点的右树。(本人采用递归的方法是构建树)
例如:
中序数组为: int arr[]={2,3,4,5,6,7,8};
后序数组为:int[] orderarr={2,4,3,6,8,7,5};
中序找到根节点index=3进行递归,左边为左子树的构建,右边为右子树的构建。
代码如下:
import java.util.Arrays;
//根据数组构建二叉树,一个数组为前序遍历一个数组为中序遍历
public static TreeNode buildTree(int[] inorder, int[] postorder) {
//数组为空时,直接返回空节点
if(inorder.length<=0||postorder.length<=0)return null;
//只有一个节点的树,直接返回该节点
if(inorder.length==1||postorder.length==1) return new TreeNode(postorder[0]);
//从后序遍历的数组中最后一个位置为该树的根节点
TreeNode tn=new TreeNode(postorder[postorder.length-1]);
//定义变量记录中序遍历中根节点的位置
int index=-1;
for(int i=0;i<inorder.length;i++){
//循环找到中序数组中的根节点位置
if(inorder[i]==postorder[postorder.length-1]){
index=i;
break;
}
}
//如果指针指向开始位置不进行左递归
if(index>=0)
//Arrays类中的copyOfRange(1,2,3)可以复制指定长度的数组,参数1表示要复制的数组,参数2表示复制数组的起始位置,参数3表示复制位置的末尾
tn.left = buildTree(Arrays.copyOfRange(inorder, 0, index), Arrays.copyOfRange(postorder,0, index));
//如果指向最后一个位置不进行右递归
if(index<=inorder.length-1)
tn.right = buildTree(Arrays.copyOfRange(inorder, index+1, inorder.length), Arrays.copyOfRange(postorder, index, postorder.length -1));
return tn;
}
**题目描述:**给出一棵树的前序遍历和中序遍历,请构造这颗二叉树
题目分析:
和上一个构建树的过程相似,找到树的根节点根据前序遍历,前序遍历的顺是:中左右,中序数组的第一个元素就是一颗树的根节点,然后在中序遍历中找到根的位置,将左边的树进行构建,再将右边的树进行构建,然后构建一颗完整的树。
代码如下:
import java.util.Arrays;
public class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
//空数组直接返回
if(preorder.length==0||inorder.length==0)return null;
//数组只有一个元素是直接为当前节点
if(preorder.length==1) return new TreeNode(preorder[0]);
//树的根节点为先序遍历数组的第一个元素
TreeNode th=new TreeNode(preorder[0]);
int index=0;//记录根节点的位置
for(int i=0;i<inorder.length;i++){
if(preorder[0]==inorder[i]){
index=i;//找到在中序数组中根节点的位置
break;
}
}
//递归传递左边的数组进行左子树的构建
if(index>=0) th.left=buildTree(Arrays.copyOfRange(preorder,1,index+1),Arrays.copyOfRange(inorder,0,index));
//递归传递右边数组进行右子树的构建
if(index<=inorder.length-1) th.right=buildTree(Arrays.copyOfRange(preorder,index+1,preorder.length), Arrays.copyOfRange(inorder,index+1,inorder.length));
return th;
}
}