剑指Offer题目:重建二叉树
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路
先序遍历顺序:根结点>左子结点->右子结点
中序遍历顺序:左子结点->根结点->右子结点
由先序遍历序列可知根结点,由中序遍历序列可知子结点相对于根结点的左右顺序,即知道哪些结点是根结点的左子树结点,哪些结点是根结点的右子树结点,这样就从中序序列中分割出了左子树序列和右子树序列,对两个子序列分别进行递归来构建树
完整代码
public class q4_reConstructBinaryTree_重建二叉树 {
private int[] pre;
private int[] in;
public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
if (pre == null || in == null || pre.length == 0 || in.length == 0 || pre.length != in.length)
return null;
this.pre = pre;
this.in = in;
return build(0, pre.length - 1, 0, in.length - 1);
}
private TreeNode build(int start_pre, int end_pre, int start_in, int end_in) {
TreeNode root = new TreeNode(pre[start_pre]);
//在中序遍历序列中找到根节点对应位置
int root_in=start_in;
while(root_in<in.length&&in[root_in]!=root.val)
root_in++;
if(root_in==in.length-1&&in[root_in]!=root.val)
return root;//找不到对应元素说明序列有问题
int leftLength=root_in-start_in;//中序遍历序列中左子序列的长度
int leftendPreorder=start_pre+leftLength;//先序遍历序列中左子序列最后的下标处
if (leftLength > 0) {
//继续构建左子树
root.left=build(start_pre+1,leftendPreorder,start_in,root_in-1);
}
if(leftLength<end_pre-start_pre)
{
//继续构建右子树
root.right=build(leftendPreorder+1,end_pre,root_in+1,end_in);
}
return root;
}
}
更多LeetCode题目及答案解析见GitHub: https://github.com/on-the-roads/LeetCode
剑指offer题目及答案解析:https://github.com/on-the-roads/SwordToOffer