重建二叉树 来源:牛客网

题目要求:

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
 
要做这道题需要深刻了解二叉树先序中序的原理,先序遍历指的是先访问此节点,然后访问此节点的左节点,在访问此节点的右节点,而中序是先访问此节点的左节点,再访问该节点,最后访问该节点的右节点。
易知先序遍历的第一个点必为根节点,而中序遍历中左节点必排在父节点之前,因此,可以用递归的方法,不断的找出根节点,代码如下:
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        int inlen=vin.size();
        if(inlen==0)
            return NULL;
        TreeNode *head=new TreeNode(pre[0]);
        int rootseri,i;
        for(i=0;vin[i]!=pre[0];i++);
        rootseri=i;
        vector <int > leftpre;vector <int > leftvin;
        for(i=0;i<rootseri;i++) 
        {
            leftpre.push_back(pre[i+1]);
            leftvin.push_back(vin[i]);
        }
        vector <int > rightpre;vector <int > rightvin;
        for(i=rootseri+1;i<inlen;i++)
        {
            rightpre.push_back(pre[i]);
            rightvin.push_back(vin[i]);
        }
        head->left=reConstructBinaryTree(leftpre,leftvin);
        head->right=reConstructBinaryTree(rightpre,rightvin);
        return head;
    }
};

 

posted @ 2018-03-12 21:13  kkdf  阅读(104)  评论(0编辑  收藏  举报