《剑指offer》-重建二叉树

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

思路：以下面的树结构进行前序和中序举例。

那么前序中的A就是最终的根节点，根据中序遍历规则，可以知道中序遍历中A左侧的节点都是A的左子树节点，A的右边就是A的右子树节点，见下图：在中序遍历中遍历，寻找‘A’节点，而A左侧的是左子树、A右侧的是右子树，而左子树的个数就是前序A后面的节点个数。

将A的左子树当做一个树结构，而重建这个树结构就是重复上一步的工作。

在中序中，寻找左子树的根节点B，重复上面的工作。循环就可以得到结果。

代码：

/**
 * 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) {
        return reConsturct(pre,0,pre.size()-1,
                           vin,0,vin.size()-1);
    }
    TreeNode* reConsturct(vector<int> pre,int start_pre,int end_pre
                         ,vector<int> in, int start_in ,int end_in)
    {
        if(start_pre > end_pre || start_in > end_in)
            return NULL;
        TreeNode* node = new TreeNode(pre[start_pre]);
        
        for(int i = start_in ; i <= end_in ; i++)
        {
            if(pre[start_pre] == in[i])
            {
                node->left = reConsturct(pre,start_pre + 1 , start_pre + i - start_in
                                        ,in ,start_in ,i -1);
                node->right = reConsturct(pre,start_pre + i - start_in + 1,end_pre
                                        , in ,i + 1 ,end_in);
            }
        }
        return node;
    }
    
};

 

来源：CSDN

作者：wangzhao93

链接：https://blog.csdn.net/wangzhao93/article/details/103606094