【力扣】106. 从中序与后序遍历序列构造二叉树

Lily-Queen

已于 2023-02-06 23:53:05 修改

阅读量46

点赞数

分类专栏：力扣文章标签： leetcode 算法职场和发展

于 2023-02-02 23:41:49 首次发布

本文链接：https://blog.csdn.net/qq_40297054/article/details/128859243

版权

力扣专栏收录该内容

20 篇文章 0 订阅

订阅专栏

递归（每层递归定义了新的vector）

如果数组长度为0，则说明是空节点
如果数组长度不为0，将后序数组最后一个元素作为节点元素
找到节点元素在中序数组中的位置，作为切割点【delimitIndex】
切割中序数组：中序左数组【[0，delimitIndex)】；中序右数组【[delimitIndex+1,end)】
切割后序数组：舍弃后序末尾元素【postorder.resize(postorder.size()-1);】；后序左数组【[0,inorderleft.size())】；后序右数组【inorderleft.size(),end】
递归左右

class Solution {
    private:
    TreeNode* traversal(vector<int>& inorder, vector<int>& postorder){
        if(postorder.size()==0) return nullptr;
        //将后序最后一个元素作为切割点
        int rootValue=postorder[postorder.size()-1];
        TreeNode* root=new TreeNode(rootValue);//根
        if(postorder.size()==1) return root;
       //在中序中找到切割点的位置
        int delimitIndex;
        for(delimitIndex=0;delimitIndex<inorder.size();delimitIndex++){
            if(inorder[delimitIndex]==rootValue) break;
        }
        //切割中序左数组和中序右数组
        vector<int> inorderleft(inorder.begin(),inorder.begin()+delimitIndex);
        vector<int> inorderright(inorder.begin()+delimitIndex+1,inorder.end());
       
        //切割后序左数组和后序右数组
        postorder.resize(postorder.size()-1);//舍去末尾元素
        vector<int> postorderleft(postorder.begin(),postorder.begin()+inorderleft.size());
        vector<int> postorderright(postorder.begin()+inorderleft.size(),postorder.end());
        //递归
        root->left=traversal(inorderleft,postorderleft);//左
        root->right=traversal(inorderright,postorderright);//右
        return root;  
    }
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if(inorder.size()==0||postorder.size()==0) return nullptr;
        return traversal(inorder,postorder);
    }
};

递归（使用下标分割数组）

中序+后序：使用后序最后一个元素作为切割点

class Solution {
    private:
    TreeNode* traversal(vector<int>& inorder,int inorderbegin,int inorderend,vector<int>& postorder,int postorderbegin,int postorderend){
        if(postorderbegin==postorderend) return nullptr;
        //将后序最后一个元素作为切割点
        int rootValue=postorder[postorderend-1];
        TreeNode* root=new TreeNode(rootValue);//根
        if(postorderend-postorderbegin==1) return root;
       //在中序中找到切割点的位置
        int delimitIndex;
        for(delimitIndex=inorderbegin;delimitIndex<inorderend;delimitIndex++){
            if(inorder[delimitIndex]==rootValue) break;
        }
        //切割中序左数组和中序右数组
        int inorderleftbegin=inorderbegin;
        int inorderleftend=delimitIndex;
        int inorderrightbegin=delimitIndex+1;
        int inorderrightend=inorderend;
       
       
        //切割后序左数组和后序右数组
        int postorderleftbegin=postorderbegin;
        int postorderleftend=postorderbegin+delimitIndex-inorderbegin;
        int postorderrightbegin=postorderbegin+(delimitIndex-inorderbegin);
        int postorderrightend=postorderend-1;
        //递归
        root->left=traversal(inorder,inorderleftbegin,inorderleftend,postorder,postorderleftbegin,postorderleftend);//左
        root->right=traversal(inorder,inorderrightbegin,inorderrightend,postorder,postorderrightbegin,postorderrightend);//右
        return root;  
    }
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if(inorder.size()==0||postorder.size()==0) return nullptr;
        return traversal(inorder,0,inorder.size(),postorder,0,postorder.size());
    }
};