题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
前序遍历:根--->左--->右;
中序遍历:左--->根--->右;
根据前序遍历和中序遍历重建二叉树的关键一步就是找到头结点,然后根据头结点区分开来左子树节点和右子数节点,然后递归左子树节点和右子树节点的前序遍历和中序遍历;
代码:
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
if(pre.size()==0) return NULL;
int rootVal=pre[0];
TreeNode* root=new TreeNode(rootVal);
vector<int> pltree,oltree;
vector<int> prtree,ortree;
int i=0;
while(i<vin.size()&&vin[i]!=rootVal){
oltree.push_back(vin[i]);
pltree.push_back(pre[i+1]);
i++;
}
i++;
while(i<vin.size()){
ortree.push_back(vin[i]);
prtree.push_back(pre[i]);
i++;
}
root->left= reConstructBinaryTree(pltree,oltree);
root->right=reConstructBinaryTree(prtree,ortree);
return root;
}
};