前序遍历:中左右
中序遍历:左中右
例:
pre:[3,9,20,15,7] in[9,3,15,20,7];
前序遍历的第一个数3,就是根节点,而中序遍历3所在的位置左边,包含在该根节点的左子树里,右边包含在右子树里,这里使用:左前,右前,左中,右中,四个区间,来表示每一子树所在的位置,如此就可以递归的构建出二叉树;
我们需要知道一个根节点的左子树和右子树的边界在前序以及中序序列中的下标,用pl, pr, il, ir来表示前序左边界、前序有边界、中序左边界、中序右边界。
图例
假设这一段前序遍历第一个数是x,找到中序遍历序列中x对应在中序遍历的下标k,中序遍历下标k左边直到il, 就是左子树在中序遍历的下标区间: [il, k - 1];同理,右子树下标区间: [k + 1, ir]
接下来寻找左子树的在前序遍历的区间下标,设点为N,由左子树区间在两个序列中长度相等有:
N - pl = k - il
N = pl + k - il
由此递归创建即可。
代码
在开始时需要使用一个hash表来装inorder的值与对应的下标,如此来建立前序点与中序点的联系。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
unordered_map<int,int>hash;
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
for (int i = 0; i < inorder.size(); i++) hash[inorder[i]] = i;
return build(preorder, inorder, 0, preorder.size() - 1, 0, inorder.size() - 1);
}
TreeNode* build(vector<int>& preorder, vector<int>& inorder, int pl, int pr, int il, int ir) {
if (pl > pr) return NULL;
int k = hash[preorder[pl]];
TreeNode* root = new TreeNode(preorder[pl]);
root->left = build(preorder, inorder, pl + 1, pl + k - il, il, k - 1);
root->right = build(preorder, inorder, pl + k - il + 1, pr,k + 1, ir);
return root;
}
};