利用中序和前序或者后序重建二叉树

二叉树数据结构

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

利用前序和中序建立二叉树

• 第一步，定义map容器，将中序in里面的元素依次与下标进行绑定，然后定义一个pre_idx整形变量，用于确定重点。
• 第二步，调用dfs函数，进行递归。 dfs(in_begin,in_end);
• 第三步，核心函数编写：
• 1.判断边界，如果当il大于ir的时候，结束。
• 2.记录下当前的根节点元素，即 pre[pre_idx];
• 3.创建新节点装下根节点，以及通过map查询到根节点所在中序的位置。
• 4.post_idx 自加一次，用于找寻下一次根节点。
• 5.进行左子树的建立：dfs(il,k-1);
• 6.进行右子树的建立：dfs(k+1,ir);
• 7.返回建立好的 二叉树。

    vector<int> pre;
    unordered_map<int,int> mp;
    int pre_idx;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        pre = preorder;
        int n = inorder.size();
        for(int i=0;i<n;i++){
            mp[inorder[i]] = i;
        }    
        return dfs(0,n-1);
    }
    TreeNode* dfs(int il,int ir){
        if(il>ir)return nullptr;
        int val = pre[pre_idx];
        int k = mp[val];
        TreeNode* root = new TreeNode(val);
        pre_idx++;
        root -> left = dfs(il,k-1);
        root -> right = dfs(k+1,ir);
        return root;
    }

利用后序和中序建立二叉树

• 第一步，定义map容器，将中序in里面的元素依次与下标进行绑定，然后定义一个post_idx整形变量，用于确定重点。
• 第二步，调用dfs函数，进行递归。 dfs(in_begin,in_end);
• 第三步，核心函数编写：
• 1.判断边界，如果当il大于ir的时候，结束。
• 2.记录下当前的根节点元素，即 post[post_idx];
• 3.创建新节点装下根节点，以及通过map查询到根节点所在中序的位置。
• 4.post_idx 自减一次，用于找寻下一次根节点。
• 5.进行右子树的建立：dfs(k+1,ir);
• 6.进行左子树的建立：dfs(il,k-1);
• 7.返回建立好的 二叉树。

    vector<int> post;
    unordered_map<int,int> mp;
    int px;

    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        int n = inorder.size();
        px = postorder.size()-1;
        post = postorder;
        for(int i=0;i<n;i++){
            mp[inorder[i]] = i;
        }
        return dfs(0,n-1);
    }
    TreeNode* dfs(int il,int ir){
        if(il>ir)return nullptr;
        int val = post[px];
        TreeNode* root = new TreeNode(val);
        int k = mp[val];
        px -- ;
        root -> right = dfs(k+1,ir);
        root -> left = dfs(il,k-1);
        return root;
    }