重建二叉树

重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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) {
        if(pre.empty() || vin.empty()) return NULL;
        return create(pre,0,pre.size()-1,vin,0,vin.size()-1);
    }
    TreeNode *create(vector<int> pre,int prel,int prer,vector<int> vin,int vinl,int vinr){
        if(prel>prer) return NULL;
        TreeNode *root=new TreeNode(pre[prel]);
        int k;
        for(k=vinl;k<=vinr;k++){
            if(vin[k]==pre[prel])
                break;
        }
        int num=k-vinl;
        root->left=create(pre,prel+1,prel+num,vin,vinl,k-1);
        root->right=create(pre,prel+1+num,prer,vin,k+1,vinr);
        return root;
    }
};

顺便，附上后序、中序建立二叉树的代码

class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.empty() || vin.empty()) return NULL;
        return create(post,0,post.size()-1,vin,0,vin.size()-1);
    }
    TreeNode *create(vector<int> post,int postl,int postr,vector<int> vin,int vinl,int vinr){
        if(postl>postr) return NULL;
        TreeNode *root=new TreeNode(post[postr]);
        int k;
        for(k=vinl;k<=vinr;k++){
            if(vin[k]==post[postr])
                break;
        }
        int numleft=k-vinl;
        root->left=create(post,postl,postl+numleft-1,vin,vinl,k-1);
        root->right=create(post,post+numleft,postr-1,vin,k+1,vinr);
        return root;
    }
};