根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 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);
}
};
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 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);
}
};
返回与给定的前序和后序遍历匹配的任何二叉树。
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);
}
};