剑指offer-刷题笔记-中难题-JZ7 重建二叉树

1.通过先序遍历找到根节点A,再通过根节点在中序遍历的位置找出左子树，右子树
2.在A的左子树中，找出左子树的根节点（先序遍历），递归进入1
3.在A的右子树中，找出右子树的根节点（先序遍历），递归进入1

class Solution {
public:
    
    TreeNode* reConstructBinaryTree(vector pre,vector vin) {
        int n = pre.size();
        int m = vin.size();
        //两个遍历都不能为0
        if(n==0||m==0)
        {
            return nullptr; 
        }
        //构建根节点
        TreeNode *root = new TreeNode(pre[0]);
        for(int i = 0;i
            //找到中序遍历中的前序第一个元素
            if(pre[0] == vin[i])
            {
                //输入pre{1，2，4，7，3，5，6，8},vin{4,7,2,1,5,3,8,6}
                //满足条件i=3, 左子树前序遍历{2，4，7},
                //左子树的前序遍历
                vector leftpre(pre.begin() + 1, pre.begin() + i + 1);//注意尾指针指到后面一个
                //左子树的中序遍历
                vector leftvin(vin.begin(),vin.begin()+i);
                //构建左子树
                root->left = reConstructBinaryTree(leftpre,leftvin);
                //右子树的前序遍历
                vector rightpre(pre.begin()+i+1, pre.end());
                //右子树的中序遍历
                vector rightvin(vin.begin()+i+1,vin.end());
                root->right = reConstructBinaryTree(rightpre, rightvin);
                break;
            }
        }   
        return root; 
    }
};

1.通过后序遍历找到根节点A,再通过根节点在中序遍历的位置找出左子树，右子树
2.在A的左子树中，找出左子树的根节点（后序遍历），递归进入1
3.在A的右子树中，找出右子树的根节点（后序遍历），递归进入1

class Solution {
public:
    
    TreeNode* reConstructBinaryTree(vector post,vector vin) {
        int n = post.size();
        int m = vin.size();
        //两个遍历都不能为0
        if(n==0||m==0)
        {
            return nullptr; 
        }
        //构建根节点
        TreeNode *root = new TreeNode(post[post.size()-1]);
        for(int i = 0;i
            //找到中序遍历中的后序最后一个元素
            if(post[post.size()-1] == vin[i])
            {
                //输入post{7,4,2,5,8,6,3,1},vin{4,7,2,1,5,3,8,6}
                //满足条件i=3, 左子树后序遍历{7，4，2},
                //左子树的后序遍历
                vector leftpost(post.begin(), post.begin() + i);//注意尾指针指到后面一个
                //左子树的中序遍历
                vector leftvin(vin.begin(),vin.begin()+i);
                //构建左子树,
                root->left = reConstructBinaryTree(leftpost,leftvin);
                //右子树的后序遍历
                vector rightpost(post.begin()+i, post.end()-1);
                //右子树的中序遍历
                vector rightvin(vin.begin()+i+1,vin.end());
                root->right = reConstructBinaryTree(rightpost, rightvin);
                break;
            }
        }   
        return root; 
    }
};