LeetCode--Construct Binary Tree from Preorder and Inorder Traversal

Given preorder and inorder traversal of a tree, construct the binary tree.

Note:

You may assume that duplicates do not exist in the tree.

思路：

这道题是树中比较有难度的题目，需要根据先序遍历和中序遍历来构造出树来。这道题看似毫无头绪，其实梳理一下还是有章可循的。下面我们就用一个例子来解释如何构造出树。

假设树的先序遍历是12453687，中序遍历是42516837。这里最重要的一点就是先序遍历可以提供根的所在，而根据中序遍历的性质知道根的所在就可以将序列分为左右子树。比如上述例子，我们知道1是根，所以根据中序遍历的结果425是左子树，而6837就是右子树。接下来根据切出来的左右子树的长度又可以在先序便利中确定左右子树对应的子序列（先序遍历也是先左子树后右子树）。根据这个流程，左子树的先序遍历和中序遍历分别是245和425，右子树的先序遍历和中序遍历则是3687和6837，我们重复以上方法，可以继续找到根和左右子树，直到剩下一个元素。可以看出这是一个比较明显的递归过程，重点在于寻找根所对应的下标。

注意：找根节点的时候i是从ileft开始，inorder[i]是根节点，i-ileft是根节点的相对位置，也就是左子树的个数。

/**
* 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:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
return buildTree(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
}
TreeNode* buildTree(vector<int>&preorder,int pleft,int pright,vector<int>&inorder,int ileft,int iright){
if(pleft>pright||ileft>iright) return NULL;
int i=0;
for(i=ileft;i<=iright;i++){
if(inorder[i]==preorder[pleft]) break;
}
TreeNode *cur=new TreeNode(preorder[pleft]);
cur->left=buildTree(preorder,pleft+1,pleft+i-ileft,inorder,ileft,i-1);
cur->right=buildTree(preorder,pleft+i+1-ileft,pright,inorder,i+1,iright);
return cur;
}
};