剑指 Offer 07. 重建二叉树（中等）

思路：

树，就想到深度优先（回溯算法）

 

一整个过程是这样的：

遍历先序数组，为每个根节点的值创建节点，并递归将node.left、node.right接上

 

代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
	//要通过先序的顺序获得根节点
	int[] preorder;
	HashMap<Integer,Integer> map=new HashMap<>();
	
    public TreeNode buildTree(int[] preorder, int[] inorder) {
		this.preorder=preorder;
		//将中序所有值的下标存入哈希
		for(int i=0;i<inorder.length;i++){
			//是以值为key，下标为value
			map.put(inorder[i],i);
		}
		return res(0,0,inorder.length-1);
    }
	//root代表先序的root，left和right都代表中序的left和right
	TreeNode res(int root,int left,int right){
		if(left>right){
			//二叉树中left不可能>right（下标）
			return null;
		}
		//将先序数组的根节点的值新建为一个节点
		//所以递归的root参数必须是先序的根节点下标
		TreeNode node=new TreeNode(preorder[root]);
		
		//获得中序数组的根节点下标
		//先通过先序获得根节点的值，再通过哈希get(value)获得根节点下标
		int i=map.get(preorder[root]);
		
		//左子树递归
		//left到i-1是左子树的范围
		node.left=res(root+1,left,i-1);
		
		//右子树递归
		//这里i-left代表用中序数组算出的左子树的长度
		//root+(i-left)+1表示先序root下标+(通过中序数组算出)左子树长度+1
		node.right=res(root+(i-left)+1,i+1,right);
		return node;
		
	}
}

 

分解：

1）因为要遍历的是先序数组，则将先序数组放到函数外面，变为全局变量

2）递归时，传入的参数有：

    i）先序的root下标

    ii）中序的左子树下标

    iii）中序的右子树下标

右子树递归时，先序的根节点不好确定：

root+(i-left)+1表示先序的根节点下标+左子树长度+1

则i表示中序数组中获得的给先序数组的根节点的下标，i-left就表示左子树的长度

 

可参照下图：

在图中，左子树的长度 = (i - left)=(中序遍历中root的下标值 - 中序遍历中的左边界) = 3 - 2 = 1 ，即为中序遍历中的[1],由于前序遍历是“中左右”的顺序，而中序遍历是“左中右”的顺序，因此中序遍历中右子树根节点=(i - left + root + 1) = (root + i - left + 1) = (前序遍历中root的下标值 + 左子树的长度 + 偏移量1) = 2 + 1 + 1 =4

 

3）哈希map存储中序数组的每一个值的下标

map.put(value,index)

 

4）递归终止条件是，遍历到左子树为空了，或右子树为空

也就是left>right

 

时间、空间复杂度分析：

时间复杂度：

    i）O（N） 将中序数组的每个值放入哈希表中需要N次遍历

    ii）O（N）每一层递归只需要查询O（1）的时间复杂度，有n个值，则需要O（N）

空间复杂度：

    i）O（N）数组保存了N个元素

    ii）O（N）树的递归最差可能是左倾或者右倾，最差会有N层，则O（N），

         最好是满二叉树，O（logN）