横断バイナリツリーは、ツリーのシーケンスの先行順走査に従って構成されています。
注意:
あなたは木が要素と重複しないと仮定することができます。
例えば、与えられました
先行順走査のプレオーダー= [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
アイデア:先行順走査がルーツについてですが、先行順が残っています各ルート・ノードは、内部を通過する前ために発見され、右し、各狭ルートを見つけるステップ(再帰
class Solution {
public:
unordered_map<int,int> pos;//目的是快速找到一个值在inorder内的下标
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int n=preorder.size();
for(int i=0;i<n;++i){
pos[inorder[i]]=i;
}
return dfs(preorder,inorder,0,n-1,0,n-1);
}
TreeNode* dfs(vector<int>& pre,vector<int>& in,int pl,int pr,int il,int ir){
if(pl>pr)return NULL;//进行到找不到结点的时候就返回
int len=pos[pre[pl]]-il;
auto root = new TreeNode(pre[pl]);
root->left=dfs(pre,in,pl+1,pl+len,il,il+len-1);
root->right=dfs(pre,in,pl+len+1,pr,il+len+1,ir);
return root;
}
};