剑指offer 07. 重建二叉树 Java版 题目分析及思路

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出:

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

   3
 /   \
9    20
    /  \
   15   7

思路一：递归

先序遍历：根节点→左子树→右子树。
中序遍历：左子树→根节点→右子树。
后续遍历：左子树→右子树→根节点。

构建前序遍历数组preorder和中序遍历数组inorder，那么我们需要构建左子树与右子树，前序遍历的第一个为根节点，找到中序遍历中根节点的位置，那么中序遍历根节点左边的数就是左子树的中序遍历数组，中序遍历根节点右边的数就是右子树的中序遍历数组，那么只需要找到左子树的前序数组和右子树的前序数组即可。通过index的值可知，由中序遍历，左子树个数为index个，那么左子树前序遍历的最后一个位置应该是原preorder数组的index，copyOfRange(a,b)方法取得到前面的a，取不到b，所以左子树前序遍历区间为(preorder, 1, index + 1)。同理，右子树的前序遍历区间为(inorder, index + 1, inorder.length))。

class Solution{
	public TreeNode buildTree(int[] preorder, int[] inorder){
		if(preorder == null || preorder.length == 0)
			return null;
		//1. 获取到树根节点的value值
		TreeNode root = new TreeNode(preorder[0]);
		
		//找到根节点的index值
		int index = findIndex(preorder, inorder);

		//2. 构建左子树与右子树
		//root.left = buildTree(左子树的前序数组，左子树的中序数组)
		root.left = buildTree(Arrays.copyOfRange(preorder, 1, index + 1), Arrays.copyOfRange(inorder, 0, index));
		//root.right = buildTree(右子树的前序数组，右子树的中序数组)
		root.right = buildTree(Arrays.copyOfRange(preorder, index + 1, preorder.length), Arrays.copyOfRange(inorder, index + 1, inorder.length));
		return root;
}
	//构造一个找到根节点位置的index函数
	public int findIndex(int[] preorder, int[] inorder){
		for(int i = 0; i< inorder.length; i++){
			if(inorder[i] == preorder[0]) 
			return i;
		}
	}
}

递归的思路比较好理解，但是占用时长较长，内存消耗也比较大。

思路二：指针

使用指针思想解决，每次遍历的时候通过指针来固定左子树和右子树在数组中的范围。使用3个指针，一个是preStart，它表示的是前序遍历开始的位置，一个是inStart，它表示的是中序遍历开始的位置。一个是inEnd，它表示的是中序遍历结束的位置，我们主要是对中序遍历的数组进行拆解。左子树指针的变化较简单，右子树的变化preStart=preStart+(index-instart+1)；preStart是当前节点比如m先序遍历开始的位置，index-instart+1就是当前节点m左子树的数量加上当前节点的数量，所以preStart+(index-instart+1)就是当前节点m右子树前序遍历开始的位置。

代码：

public TreeNode buildTree(int[] preorder, int[] inorder) {
    return helper(0, 0, inorder.length - 1, preorder, inorder);
}

public TreeNode helper(int preStart, int inStart, int inEnd, int[] preorder, int[] inorder) {
    if (preStart > preorder.length - 1 || inStart > inEnd) {
        return null;
    }
    //创建结点
    TreeNode root = new TreeNode(preorder[preStart]);
    int index = 0;
    //找到当前根节点在中序遍历中的位置，然后再把数组分两部分
    for (int i = inStart; i <= inEnd; i++) {
        if (inorder[i] == root.val) {
            index = i;
            break;
        }
    }
    root.left = helper(preStart + 1, inStart, index - 1, preorder, inorder);
    root.right = helper(preStart + index - inStart + 1, index + 1, inEnd, preorder, inorder);
    return root;
}

参考资料：bilibili@mmmmm8811、微信公众号：数据结构与算法