剑指Offer-重建二叉树

2017/3/9 posted in  算法-数据结构  

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

结题思路

在二叉树的前序遍历中,第一个数字总是数的根节点的值。但在中序遍历中,根节点的值在序列的中间,左子树的结点的值位于根节点的左边,而右子树的节点的值位于根节点的值的右边,因此,我们需要扫描中序遍历,才能找到根节点的值。实例如图:
 -c

既然我们已经分别找到了左、右子树的前序遍历和中序遍历序列,我们可以用同样的方法去分别构建左右子树。也就是递归去实现。

代码

/**
 * 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 reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length);
    }
    
    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]);
        int k = find(in, pre[pStart], iStart, iEnd) - iStart;
        node.left = reConstructBinaryTree(pre, pStart+1, pStart+k, in, iStart, iStart+k);
        node.right = reConstructBinaryTree(pre, pStart+k+1, pEnd, in, iStart+k+1, iEnd);
        return node;
        
    }
    
    public int find(int[] array,int target,int start,int end){
        for(int i = start;i<=end;i++){
            if(array[i] == target){
                return i;
            }
        }
        return -1;
    }
}