2021-06-23

重建二叉树

https://www.cnblogs.com/edisonchou/p/4741099.html

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

 TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) 
    {
        return reBuild(pre,0,pre.size()-1,vin,0,vin.size()-1);
    }
    
    
    TreeNode* reBuild(vector<int> &preOrder,int startPreOrder,int endPreOrder,vector<int>&inOrder,int startInOrder,int endInOrder)
    {
        if(startPreOrder>endPreOrder)
            return NULL;
        
        //前序遍历的第一个数是根节点的值
        int rootValue=preOrder[startPreOrder];
        TreeNode* root=new TreeNode(rootValue);
        
        //在中序遍历中找到根结点的下标
        int rootInOrder=startInOrder;
        while(rootInOrder<=endInOrder && inOrder[rootInOrder]!=rootValue)
        {
            rootInOrder++;
        }
        
        //构建左子树
        root->left=reBuild(preOrder, startPreOrder+1, startPreOrder+rootInOrder-startInOrder, 
                           inOrder, startInOrder, rootInOrder-1);
        //构建右子树
        root->right=reBuild(preOrder, startPreOrder+rootInOrder-startInOrder+1, endPreOrder,
                            inOrder, rootInOrder+1,endInOrder);
        
        return root;
    }