剑指offer_4 重建二叉树

来源:互联网 发布:深圳网络布线 编辑:程序博客网 时间:2024/06/05 20:27

题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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 Test {    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {        TreeNode node=reConstructBinaryTree(pre, 0, pre.length-1,in,0,in.length-1);        return node;    }    public TreeNode reConstructBinaryTree(int[] pre,int pStart,int pEnd,int[] in,int iStart,int iEnd){        if(pStart>pEnd ||iStart>iEnd)            return null;        TreeNode node=new TreeNode(pre[pStart]);        for (int i = iStart; i <=iEnd; i++) {            if(in[i]==pre[pStart]){//在in中找到pre中所表示的父结点            //左子树肯定是pre的下一个结点开始而且是in中i的左边部分                node.left=reConstructBinaryTree(pre, pStart+1, pEnd, in, iStart, i-1);//右子树为in中i的右边部分,为pre中(i-iSatrt+1)+pStart的位置开始的,其中i-iStart+1表示所找到的结点是in中的第几个结点,而在pre当中的话,从pStart开始往后数几个点开始就属于右子树的范围。                node.right=reConstructBinaryTree(pre, i-iStart+pStart+1, pEnd, in,i+1,iEnd);            }        }        return node;    }}class TreeNode {     int val;     TreeNode left;     TreeNode right;     TreeNode(int x) { val = x; }}