从先序与中序遍历序列构造二叉树(二叉树与递归)
1、根据一棵树的前序遍历与中序遍历构造二叉树。
2、假设树中没有重复的元素。
/**
* 从前序遍历和中序遍历构造一颗二叉树:递归
* @param preorder
* @param inorder
* @return
*/
public TreeNode buildTree(int[] preorder, int[] inorder) {
//递归终止条件
if(preorder.length == 0 || inorder.length == 0){
return null;
}
//递归内容
//根据先续遍历的第一个元素确定根节点
TreeNode root = new TreeNode(preorder[0]);
//从中序遍历中查找值等于根节点值得索引下表index
int index = -1;
for(int i = 0 ;i<inorder.length;i++){
if(inorder[i] == root.val){
index = i;
break;
}
}
//确定index后,将中序数组分为:[0-index)为左子树,[index+1-endindex]为右子树两部分
//先序数组分为[1-index+1)为左子树,[index+1-endindex]为右子树两部分
//Arrays.copyOfRange(inorder,from,to); from:包含,to:不包含
int[] inLeft = Arrays.copyOfRange(inorder,0,index);
int[] inRight = Arrays.copyOfRange(inorder,index+1,inorder.length);
int[] preLeft = Arrays.copyOfRange(preorder,1,index+1);
int[] preRight = Arrays.copyOfRange(preorder,index+1,preorder.length);
//递归处理中序遍历的左边和先序遍历的左边
root.left=buildTree(preLeft,inLeft);
//递归处理中序遍历的右边和先序遍历的右边
root.right=buildTree(preRight,inRight);
return root;
}
public class TreeNode{
int val;
TreeNode left,right;
public TreeNode(int val){this.val = val;}
}
/**
* 从前序遍历和中序遍历构造一颗二叉树:递归
* @param preorder
* @param inorder
* @return
*/
public TreeNode buildTree2(int[] preorder, int[] inorder) {
if(preorder.length == 0 || inorder.length == 0){
return null;
}
return dfs2(preorder,inorder);
}
private TreeNode dfs2(int[] preorder, int[] inorder){
if(inorder.length ==0){
return null;
}
//递归内容
//根据先续遍历第一个元素确定根节点
TreeNode root = new TreeNode(preorder[0]);
//从中序遍历中查找值等于根节点值得索引下表index
int index = -1;
for(int i = 0 ;i<inorder.length;i++){
if(inorder[i] == root.val){
index = i;
break;
}
}
//确定index后,将中序数组分为:[0-index)为左子树,[index+1-endindex]为右子树两部分
//先序数组分为[1-index+1)为左子树,[index+1-endindex]为右子树两部分
//Arrays.copyOfRange(inorder,from,to); from:包含,to:不包含
int[] inLeft = Arrays.copyOfRange(inorder,0,index);
int[] inRight = Arrays.copyOfRange(inorder,index+1,inorder.length);
int[] preLeft = Arrays.copyOfRange(preorder,1,index+1);
int[] preRight = Arrays.copyOfRange(preorder,index+1,preorder.length);
//递归处理中序遍历的左边和先序遍历的左边
root.left=dfs2(preLeft,inLeft);
//递归处理中序遍历的右边和先序遍历的右边
root.right=dfs2(preRight,inRight);
return root;
}