题目:输入二叉树的前序遍历和中序遍历序列,重构其二叉树,假设输入的数字都是不重复的。
例如前序序列是 {1,2,4,7,3,5,6,8},中序序列是{4,7,2,1,5,3,8,6}。
分析如下图所示:
算法步骤:
- 前序序列pre的第一个元素即根节点。
- 遍历中序序列mid,找到的根节点,得到左子树的大小left_size。
- 左子树的中序序列为mid[0,left_size ),右子树的中序序列为mid[left_size+1, mid.size)
- 左子树的前序序列为pre[1,left_size ),右子树的前序序列为pre[left_size+1, pre.size)
代码如下:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
if (pre.size() == 0 || vin.size() == 0) {
return nullptr;
}
int root_val = pre[0];
TreeNode* root = new TreeNode(root_val);
int left_size = 0;
for (int i = 0; i < vin.size(); ++i) {
if (vin[i] == root_val) {
left_size = i;
break;
}
}
vector<int> left_vin;
left_vin.assign(vin.begin(), vin.begin() + left_size);
vector<int> right_vin;
right_vin.assign(vin.begin() + left_size + 1, vin.end());
vector<int> left_pre;
left_pre.assign(pre.begin() + 1, pre.begin() + 1 + left_size);
vector<int> right_pre;
right_pre.assign(pre.begin() + 1 + left_size, pre.end());
root->left = reConstructBinaryTree(left_pre, left_vin);
root->right = reConstructBinaryTree(right_pre, right_vin);
return root;
}