重建二叉树

题目描述

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

preorder = [3,9,20,15,7]
inorder =  [9,3,15,20,7]

 

解题思路

前序遍历的第一个值为根节点的值，使用这个值将中序遍历结果分成两部分，左部分为树的左子树中序遍历结果，右部分为树的右子树中序遍历的结果。

算法代码如下，注意hashmap的使用：

import java.util.HashMap;

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




public class Order {
	
	private static HashMap<Integer,Integer> map=new HashMap<>();
	
	public TreeNode construct(int[] pre,int preL, int preR, int inL)
	{
		if(preL>preR)
		{
			return null;
		}
		
		TreeNode root=new TreeNode(pre[preL]);
		int index=map.get(root.val);
		int leftSize=index-inL;
		root.left=construct(pre,preL+1,preL+leftSize,inL);
		root.right=construct(pre,preL+leftSize+1,preR,inL+leftSize+1);
		return root;
		
	}
	
	
	public TreeNode reConstructBinaryTree(int [] pre,int [] in) 
	{
		int length=in.length;
		for(int i=0;i<length;i++)
		{
			map.put(in[i], i);
		}
		
		return construct(pre,0,length-1,0);
    }

	public static void main(String[] args) {
		
		

	}

}