题目
输入一棵二叉树前序遍历和中序遍历的结果,请重建该二叉树。
注意:
二叉树中每个节点的值都互不相同;
输入的前序遍历和中序遍历一定合法;
样例
给定:
前序遍历是:[3, 9, 20, 15, 7]
中序遍历是:[9, 3, 15, 20, 7]
返回:[3, 9, 20, null, null, 15, 7, null, null, null, null]
返回的二叉树如下所示:
3
/ \
9 20
/ \
15 7
方法一:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
unordered_map<int, int> pos;
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int n=inorder.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>& preorder, vector<int>& inorder, int pl, int pr, int il, int ir) { //pl、pr表示的是前序遍历的左右指针;指向第一和最后元素
//il和il相同
if((pl > pr) && (il > ir)) return NULL;
int k=pos[preorder[pl]] - il; //由于pos表示的inorder中各个元素位置;所以这里的k是根节点左侧有多少个元素。
TreeNode* root = new TreeNode(preorder[pl]);
root->left = dfs(preorder, inorder, pl+1, pl+k, il, il+k-1);
root->right = dfs(preorder, inorder, pl+k+1, pr, il+k+1, ir);
return root;
}
};
心得
-
中心在于转化为个数;个数联动了前序和后序遍历。 而后面
root->left = dfs(preorder, inorder, pl+1, pl+k, il, il+k-1); root->right = dfs(preorder, inorder, pl+k+1, pr, il+k+1, ir);
便是基于个数而写出的。 根节点左子树前序是从pl+1到pl+k的数;同理左子树中序和右子树。