剑指Offer——面试题6：重构二叉树

题目描述

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

分析

在二叉树前序遍历序列中，第一个数字总是树的根节点的值。但在中序遍历中，根节点的值在序列中间。所有的左子树都位于根节点的左边，同理所有的右子树都在根节点的右边。

例如

由于在中序遍历中，有3个数字是左子树结点的值，因此可以知道左子树总共有3个左子结点。同样，在前序遍历的序列中，根节点后面的3个数字就是3个左子树结点的值，再后面就是右子数的值。

这样我们就分别找到了左右子数对应的子序列。

我们已经分别找到了左右子数的前序和中序遍历，那么我们可以用同样的方法分别去构建左右子树。

实现

public class T06 {

	public TreeNode reConstructBinaryTree(int [] pre, int [] in) {
		TreeNode root = reConstructBinaryTree(pre, 0, pre.length-1, in, 0, in.length-1);
		return root;
	}

	public TreeNode reConstructBinaryTree(int [] pre, int startPre, int endPre, int [] in, int startIn, int endIn) {
		
		if (startPre > endPre || startIn > endIn) {
			return null;
		}
		
		TreeNode root = new TreeNode(pre[startPre]);
		
		for (int i = startIn; i <= endIn; i++) {
			
			if (in[i] == pre[startPre]) {
				root.left = reConstructBinaryTree(pre, startPre + 1, startPre + i - startIn, in, startIn, i - 1);
				root.right = reConstructBinaryTree(pre, i - startIn + startPre + 1, endPre, in, i + 1, endIn);
			}
		}
		
		return root;
	}
}


class TreeNode {
	
	int val;
	TreeNode left;
	TreeNode right;
	
	TreeNode(int x) { 
		val = x; 
	}
}