描述
给定节点数为 n 的二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序{4,7,2,1,5,3,8,6},则重建出如下图所示。

输入:
[1,2,4,7,3,5,6,8],[4,7,2,1,5,3,8,6]
返回值:
{1,2,3,4,#,5,6,#,7,#,#,8}
说明:
返回根节点,系统会输出整颗二叉树对比结果,重建结果如题面图示
思路:知道前序遍历和中序遍历,那么在前序遍历中知道根节点之后,能在中序遍历中划分出左子树和右子树,分别对左子树和右子树递归进行这一过程即可

/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
map<int, int>map;
//map中存放中序遍历元素的下标
TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin) {
//判断是否为空
if (pre.size() == 0)
return nullptr;
for (int i = 0; i < vin.size(); i++)
map[vin[i]] = i;
TreeNode* root = f(pre, 0, pre.size() - 1, vin, 0, vin.size() - 1);
return root;
}
TreeNode* f(vector<int>pre, int l1, int r1, vector<int>vin, int l2, int r2) {
if (l1 > r1 && l2 > r2)
return nullptr;
TreeNode* root = new TreeNode(pre[l1]);
//取出当前前序遍历中的根节点在中序遍历中的下标
int i = map[pre[l1]];
//进行划分
root->left = f(pre, l1 + 1, l1 + i - l2, vin, l2, i - 1);
root->right = f(pre, l1 + i - l2 + 1, r1, vin, i + 1, r2);
return root;
}
};

被折叠的 条评论
为什么被折叠?



