构建二叉树

105. 从前序与中序遍历序列构造二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解:核心代码

int L=pos-x;     

pos是根结点在中序数组(inorder)中的位置;L是当前根结点的左子树包含结点的个数

 

dfs(preorder, inorder, l, r, x, y); 

preorder前序遍历数组,inorder是中序遍历数组,l与r是前序遍历当前根结点左子树的结点集合,x与y是中序遍历序遍历当前根结点左子树的结点集合

 

rt->left=dfs(preorder,inorder,l+1,l+L,x,pos-1);     

rt->right=dfs(preorder,inorder,l+L+1,r,pos+1,y);

preoder:    3      9      20      15      7  
            l     l+1    l+L+1            r
                  l+L                    
            
inorder:    9      3      15      20      7
            x     pos    pos+1            y
            
红字代表左子树结点集合  蓝字代表右子树结点集合
class Solution {
public:
    TreeNode* dfs(vector<int>& preorder, vector<int>& inorder,int l,int r,int x,int y){
        if(l>r||x>y) return NULL;
        if(l==r||x==y) return new TreeNode(preorder[l]);
        TreeNode* rt = new TreeNode(preorder[l]);
        int pos=0;
        for(int i=x;i<=y;i++){
            if(inorder[i]==preorder[l]){
                pos=i;
                break;
            }
        }
        int L=pos-x;
        rt->left=dfs(preorder,inorder,l+1,l+L,x,pos-1);
        rt->right=dfs(preorder,inorder,l+L+1,r,pos+1,y);
        return rt;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int len1=preorder.size();
        int len2=inorder.size();
        if(!len1||!len2) return NULL;
        return dfs(preorder, inorder, 0, len1-1, 0, len2-1);
    }
};

 

106. 从中序与后序遍历序列构造二叉树

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如,给出

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解:核心代码

int L=pos-l-1;
rt->left=dfs(inorder,postorder,l,pos-1,x,x+L);
rt->right=dfs(inorder,postorder,pos+1,r,x+L+1,y-1);

class Solution {
public:
    TreeNode* dfs(vector<int>& inorder, vector<int>& postorder,int l,int r,int x,int y){
        if(x>y) return NULL;
        if(x==y) return new TreeNode(postorder[y]);
        TreeNode* rt=new TreeNode(postorder[y]);
        int pos=0;
        for(int i=l;i<=r;i++){
            if(inorder[i]==postorder[y]){
                pos=i;
                break;
            }
        }
        int L=pos-l-1;
        rt->left=dfs(inorder,postorder,l,pos-1,x,x+L);
        rt->right=dfs(inorder,postorder,pos+1,r,x+L+1,y-1);
        return rt;
    }
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        int len1=inorder.size();
        int len2=postorder.size();
        if(!len1||!len2) return NULL;
        return dfs(inorder,postorder,0,len1-1,0,len2-1);
    }
};

 

889. 根据前序和后序遍历构造二叉树

返回与给定的前序和后序遍历匹配的任何二叉树。

 pre 和 post 遍历中的值是不同的正整数。

示例:

输入:pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
输出:[1,2,3,4,5,6,7]
 

提示:

1 <= pre.length == post.length <= 30
pre[] 和 post[] 都是 1, 2, ..., pre.length 的排列
每个输入保证至少有一个答案。如果有多个答案,可以返回其中一个。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解:核心代码

int L=pos-x;
rt->left=dfs(pre,post,l+1,l+1+L,x,pos);
rt->right=dfs(pre,post,l+2+L,r,pos+1,y-1);

class Solution {
public:
    TreeNode* dfs(vector<int>& pre, vector<int>& post,int l,int r,int x,int y){
        if(l>r) return NULL;
        if(l==r) return new TreeNode(pre[l]);
        TreeNode* rt=new TreeNode(pre[l]);
        int pos=0;
        for(int i=x;i<=y;i++){
            if(post[i]==pre[l+1]){
                pos=i;
                break;
            }
        }
        int L=pos-x;
        rt->left=dfs(pre,post,l+1,l+1+L,x,pos);
        rt->right=dfs(pre,post,l+2+L,r,pos+1,y-1);
        return rt;
    }
    TreeNode* constructFromPrePost(vector<int>& pre, vector<int>& post) {
        int len1=pre.size();
        int len2=post.size();
        if(!len1||!len2) return NULL;
        return dfs(pre, post, 0, len1-1, 0, len2-1);
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值