剑指Offer(牛客网 Java)04——重建二叉树

题目描述

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

解题思路

树的前序遍历中，第一个值一定是树的根结点的值；
中序遍历中，根结点左部分就是树中根结点的左子树的中序遍历，而右部分则是根结点右子树的中序遍历。

根据前序遍历中第一个值，就能将中遍历拆分成左右子树的中序遍历，根据中序遍历的划分，同样可以将先序遍历拆分成左右子树的前序遍历，这样一来，就只需要将根节点的左右两个子树重建好就行了，这不就是递归嘛！

1. 根据当前前序序列的第一个结点确定根结点；
2. 找到当前根节点在中序遍历序列中的位置，记为mid；
3. 拆分左右子树，0 ~ mid - 1是左子树， mid + 1 ~ in.length是右子树；同样根据长度将先序遍历也拆分了，1 ~ mid是左子树，mid + 1 ~ pre.length；
4. 拆分后对子树分别使用同样的方法分解

代码实现

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.Arrays;
public class Solution {
	public TreeNode reConstructBinaryTree(int [] pre, int [] in) {
	     int pLen = pre.length;
	     int iLen = in.length;
	     
	     TreeNode headNode = null;
	     
	     if (pLen > 0 && iLen > 0) {
	         headNode = binaryTree(pre, in);
	     }
	     
	     return headNode;
	 }
	
	 private TreeNode binaryTree(int[] pre, int[] in) {
	     int pLen = pre.length;
	     int iLen = in.length;
	     
	     //pre为空，表示当前子树重建结束
	     if (0 == pLen) {
	         return null;
	     }
	     
	     //创建新的根结点
	     int curValue = pre[0];
	     TreeNode newNode = new TreeNode(curValue);
	     
	     //当前结点是叶子结点
	     if (1 == pLen) {
	         return newNode;
	     }
	     
	     //获取根节点在中序遍历中的位置
	     int mid = getIndex(in, curValue);
	     
	     if (-1 != mid) {
     	     //如果中序遍历中存在根节点
			 //重建左子树
			 newNode.left = binaryTree(Arrays.copyOfRange(pre, 1, mid + 1), Arrays.copyOfRange(in, 0, mid));
	     }
	     
	     if (mid + 1 < pLen) {
	     	 //如果有右子树
	     	 //重建右子树
	         newNode.right = binaryTree(Arrays.copyOfRange(pre, mid + 1, pLen), Arrays.copyOfRange(in, mid + 1, iLen));
	     }
	     
	     return newNode;
	 }
	
	 private int getIndex(int[] in, int target) {
	     int iLen = in.length;
	     for (int curIndex = 0 ; curIndex < iLen ; curIndex++) {
	         if (target == in[curIndex]) {
	             return curIndex;
	         }
	     }
	     
	     return -1;
	 }
}