剑指 Offer（第2版）面试题 7：重建二叉树

剑指 Offer（第2版）面试题 7：重建二叉树

题目来源：18. 重建二叉树

解法1：递归

递归建立整棵二叉树：先递归创建左右子树，然后创建根节点，并让指针指向两棵子树。

1. 先利用前序遍历找根节点：前序遍历的第一个数，就是根结点的值。
2. 在中序遍历序列中找到当前根结点的位置 k，则在序列中 k 的左边是左子树的中序遍历序列（in_left, in_left + k - 1），右边是右子树的中序遍历序列（in_left + k + 1, in_right）。
3. 假设左子树的中序遍历的长度是 l，则在前序遍历中，根结点后面的 l 个数就是左子树的前序遍历序列（pre_left + 1, pre_left + k），剩下的数是右子树的前序遍历序列（pre_left + k + 1, pre_right）。
4. 获得左右子树的前序遍历序列和中序遍历序列后，我们可以先递归创建出左右子树，然后再创建根结点。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution
{
public:
	// 记录每个结点在中序遍历序列中的位置
	unordered_map<int, int> pos;
	// 根据前序遍历序列和中序遍历序列，构建二叉树
	TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder)
	{
		int n = preorder.size();
		// 记录每个结点在中序遍历序列中的位置
		for (int i = 0; i < n; i++)
			pos[inorder[i]] = i;
		return dfs(preorder, inorder, 0, n - 1, 0, n - 1);
	}
	// 辅函数 - 递归遍历二叉树
	TreeNode *dfs(vector<int> &pre, vector<int> &in, int pre_left, int pre_right, int in_left, int in_right)
	{
		// 注意递归结束条件
		if (pre_left > pre_right)
			return nullptr;
		// k 表示当前根结点在所给的中序遍历范围区间内的位置（分割左右子树）
		int k = pos[pre[pre_left]] - in_left;
		// 左子树的中序遍历序列（in_left ~ in_left + k - 1），右子树的中序遍历序列（in_left + k + 1 ~ in_right）
		// 左子树的前序遍历序列（pre_left + 1 ~ pre_left + k），右子树的前序遍历序列（pre_left + k + 1, pre_right）
		TreeNode *root = new TreeNode(pre[pre_left]);
		root->left = dfs(pre, in, pre_left + 1, pre_left + k, in_left, in_left + k - 1);
		root->right = dfs(pre, in, pre_left + k + 1, pre_right, in_left + k + 1, in_right);
		return root;
	}
};

复杂度分析：

时间复杂度：O(n)，其中 n 是数组 preorder 和 inorder 的长度。

空间复杂度：O(n)，其中 n 是数组 preorder 和 inorder 的长度。