剑指Offer（树）-重建二叉树

 

　　（重建二叉树）题目描述：

---

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

---

　　解题思路：寻找根节点，重组左子树和右子树

　　已知二叉树的前序遍历的方式，所以输入前序遍历序列的第一个元素就是该二叉树的根节点的val值，然后可以从中序遍历的序列中取左子树，此时就是重新递归调用定义的方法reConstructBinaryTree(),里面的参数放根节点后的（前序遍历结果）和根节点前的（中序遍历的结果），这样子得到的就会返回一个此左子树的根节点，也就是root.left，同理也可以得到rtoot.right。

package dsaa.sword.根据前序和后序重构二叉树;

import java.util.Arrays;

class TreeNode {
	int val;
	TreeNode left;
	TreeNode right;

	public TreeNode(int val) {
		super();
		this.val = val;
	}
}

public class Solution {
	/**
	 * 采用递归的思想
	 * 
	 */
	public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
		// 先进行空判断
		if (pre.length == 0 || in.length == 0) {
			return null;
		}
		// 前序遍历的第一个元素就是根节点的值
		TreeNode root = new TreeNode(pre[0]);
		// 从中序遍历的结果中去寻找原树的左右子树的大小
		for (int i = 0; i < in.length; i++) {
			if (root.val == in[i]) {
				// 找到中序数组中和根节点的值相同的元素之后，其实此时的i的值就代表了原树的左子树的大小
				root.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i + 1), Arrays.copyOfRange(in, 0, i));
				root.right = reConstructBinaryTree(Arrays.copyOfRange(pre, i + 1, pre.length),
						Arrays.copyOfRange(in, i + 1, in.length));
			}
		}
		return root;
	}
}

　　

　　这里需要重点说明下Arrays.copyOfRange方法，它的三个参数分别是“从哪里取”，from（取得到），to（取不到）。所以重点的指针索引要注意位置别出错。