重建二叉树-(递归)
今天遇到一道比较难实现的题,参考别人的代码进行学习总结下…
总结:感觉难点在于如何实现代码,整体思路想明白了,但是要真的实现却发现困难,主要在于传入参数的定义,将两个序列分离开来进行递归查询,思路还是比较新奇的。
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
return returnTreeNode(0,pre.length-1,pre,0,in.length-1,in);
}
/**
ps pe表示前序遍历的开头结尾
is ie表示中序遍历的开头结尾
*/
private TreeNode returnTreeNode(int ps,int pe,int[] pre,int is,int ie,int[] in){
if(ps>pe||is>ie)
return null;
TreeNode root = new TreeNode(pre[ps]);
int i=is;
while(i<in.length&&pre[ps]!=in[i]){
++i;
}
root.left = returnTreeNode(ps+1,i-is+ps,pre,is,i-1,in);
root.right = returnTreeNode(i-is+ps+1,pe,pre,i+1,ie,in);
return root;
}
}