输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
限制:
0 <= 节点个数 <= 5000
为什么需要已知前序遍历和中序遍历??
因为只知道前序遍历或只知道中序遍历,是无法确定左子树、根、右子树的。因为前序遍历顺序为根左右,中序遍历为左根右。
思路
前序遍历数组的从左往右是一系列顺序排列的节点
找出根节点,得到根节点的val
从前序遍历中取出一个数,再去中序遍历中寻找相同的数
,该数将中序遍历数组分割成了左右子树。
递归以上步骤,直到达到递归终止条件。
前序遍历的顺序
从根结点开始,进入左子树下层,再进入左子树,再进入左子树…直到左子树为空,再考虑空左子树的兄弟右子树。即使进入了兄弟右子树,也还是继续探索该右子树的左子树…
中序遍历的顺序
就是一条垂直线从左往右扫描二叉树(犹如一根垂直线扫过一个正三角形)。
/**
* 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:
map<int, int> indexTable; // 方便直接获取inorder数组的索引
int p = 0; // 用于遍历整个preorder数组,使用全局变量,再递归时更新。
// 本次递归中,以preorder[p]为根的二叉树的区间范围就是inorder[istart,iend),是一个左闭右开的区间,注意边界判断。
TreeNode* dfs(vector<int>& preorder, vector<int>& inorder, int istart, int iend){
// printf("p=%d, istart=%d, iend=%d\n", p, istart, iend);
if(p >= preorder.size() || (istart >= iend)) return NULL;
// 创建根结点,以preorder[p]为根
TreeNode* curNode = new TreeNode(preorder[p]);
// 处理当前根结点的左子树,inorder[start, root)都属于左子树,inorder[root + 1, end)都属于右子树
// 获取根结点在inorder中的位置,inorder[root]就是根
int root = indexTable[preorder[p]];
p += 1;
curNode->left = dfs(preorder, inorder, istart, root);
// 再处理右子树
curNode->right = dfs(preorder, inorder, root + 1, iend);
return curNode;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
for (int i = 0; i < inorder.size(); ++i)
indexTable[inorder[i]] = i;
// inorder的区间是左闭右开
return dfs(preorder, inorder, 0, inorder.size());
}
};