剑指Offer 07.重建二叉树

重建二叉树题目

思路

前序遍历的顺序是[(根结点), (左子树前序遍历), (右子树前序遍历)]
中序遍历的顺序是[(左子树中序遍历), (根结点), (右子树中序遍历)]
采用递归的思想。既然前序遍历是从根结点开始，那么就根据前序遍历数组的第一个结点记为root，在中序遍历中找到root。然后删除前序遍历数组中的根结点。
以新的前序遍历数组和中序遍历数组中root左侧部分为输入，构造root的左子树。
再以新的前序遍历数组（注意此时前序遍历数组的首结点在上面的递归中删除了一个）和中序遍历数组中root右侧部分为输入，构造root的右子树。
最后返回root即可。

代码

TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
	if (preorder.size() == 0 || inorder.size() == 0)
	     return nullptr;
	 // 根据前序遍历数组获得根结点
	 TreeNode *root = new TreeNode(preorder[0]);
	 // 将其删除
	 preorder.erase(preorder.begin());
	 // 找到其在中序遍历数组中的位置
	 vector<int>::iterator i = inorder.begin();
	 // 一定能检测到，所以不用进行边界检测
	 while ((*i) != root->val)
	     i++;
	
	 vector<int> leftIn(inorder.begin(), i);
	 root->left = buildTree(preorder, leftIn);
	 vector<int> rightIn(i + 1, inorder.end());
	 root->right = buildTree(preorder, rightIn);
	 return root;
    }