重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

解题思路

本题需要利用递归的思想。首先，因为前序遍历的顺序是根节点->左子树->右子树，因此前序遍历的第一个元素就是二叉树的根节点。然后，因为中序遍历的顺序是左子树->根节点->右子树，因此在中序遍历的结果中确定了根节点的位置后，左右子树的大小也就确定了，根据前序遍历和中序遍历的结果，左右子树的前序遍历和中序遍历的结果也就确定了，从而可以进行递归，重建二叉树。

代码

/**
 * Definition for binary tree
 * 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.empty() || vin.empty())
            return NULL;
        
        TreeNode* head = new TreeNode(pre.front()); //前序遍历第一个元素一定是头节点
        
        vector<int>::iterator iter = vin.begin();
        for (; iter != vin.end(); iter++)
        {
            if (*iter == pre.front()) //查找头节点在中序遍历中的位置，从而确定左右子树
                break;
        }
        
        int lsize = iter - vin.begin(), rsize = vin.size() - lsize - 1; //左右子树的节点数
        vector<int> lpre(pre.begin() + 1, pre.begin() + 1 + lsize);
        vector<int> rpre(pre.begin() + 1 + lsize, pre.end());
        vector<int> lvin(vin.begin(), vin.begin() + lsize);
        vector<int> rvin(vin.begin() + lsize + 1, vin.end());
        
        head->left = reConstructBinaryTree(lpre, lvin);
        head->right = reConstructBinaryTree(rpre, rvin);
        
        return head;

    }
};