重建二叉树

题目描述

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

思路

根据二叉树的前序遍历可以知道根节点是1，在中序遍历中根节点前的为左子树{4，7，2}，根节点后的为右子树{5，3，8，6}，通过递归的方法，在分别取构建左右子树，最后将构建好的左右子树放进根节点中。

代码

	  // Definition for binary tree
	  public class TreeNode {
	      int val;
	      TreeNode left;
	      TreeNode right;
	      TreeNode(int x) { val = x; }
	  }
	  
	public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
		
		if(pre.length ==0 || in.length == 0){
			return null;
		}
//		只有一个节点
		if(pre.length == 1){
			return new TreeNode(pre[0]);
		}
//		找根节点
		int rootIndex = 0;
		int rootValue = pre[0];
		TreeNode root = new TreeNode(rootValue);
		root.left = null;
		root.right = null;
		for(int i=0;i<in.length;i++){
			if(rootValue == in[i]){
				rootIndex = i;
				break;
			}
		}
		
//		左右子树对应的前序和中序
//		包括前面不包括后面
		int[] leftPre = Arrays.copyOfRange(pre, 1, rootIndex+1);
		int[] rightPre = Arrays.copyOfRange(pre, rootIndex+1, pre.length);
		
		int[] leftIn = Arrays.copyOfRange(in, 0, rootIndex);
		int[] rightIn = Arrays.copyOfRange(in, rootIndex+1, in.length);
		
//		分别构建左右子树
		root.left = reConstructBinaryTree(leftPre, leftIn);
		root.right = reConstructBinaryTree(rightPre, rightIn);
		return root;
	}