由先序和中序重建二叉树

Posted 2022-06-07 tianzeng

思路：

　　我们知道，前序遍历的第一个节点就是树的根节点，所以我们先根据前序遍历序列的第一个数字创建根结点，接下来在中序遍历序列中找到根结点的位置，根节点的左边就是左子树，右边就是右子树，这样就能确定左、右子树结点的数量。在前序遍历和中序遍历的序列中划分了左、右子树结点的值之后，就可以递归地去分别构建它的左右子树。

/**
 * 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()||pre.size()!=vin.size())
            return nullptr;
         
        return reConstructBinaryTreeCore(pre,0,pre.size()-1,vin,0,vin.size()-1);
    
    TreeNode *reConstructBinaryTreeCore(vector<int> pre,int preStart,int preEnd,
                                        vector<int> vin,int vinStart,int vinEnd)
    
        if(preStart>preEnd||vinStart>vinEnd)
            return nullptr;
        
        TreeNode *root=new TreeNode(pre[preStart]);
        for(int i=vinStart;i<=vinEnd;++i)
        
            if(pre[preStart]==vin[i])
            
                root->left=reConstructBinaryTreeCore(pre,preStart+1,preStart+i-vinStart,vin,vinStart,i-1);
                root->right=reConstructBinaryTreeCore(pre,preStart+i-vinStart+1,preEnd,vin,i+1,vinEnd);
                break;
            
        
        return root;
    
;