从二叉树的前序遍历序列和中序遍历序列重构出二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

题目来源：牛客网

问题分析

其实根据前序和中序的特点就可以做了，前序的第一个元素必定是根节点，那么在中序的序列中查找这个根节点的位置，两侧分为两部分，递归下去即可。

实现

/**
* Definition for binary tree
* struct TreeNode {
*     int val;
*     TreeNode *left;
*     TreeNode *right;
*     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
struct TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> in) {
if (pre.empty())
return NULL;

int mid = findPos(in, pre[0]);
vector<int> left_p(pre.begin()+1, pre.begin()+mid+1), right_p(pre.begin()+mid+1, pre.end());
vector<int> left_i(in.begin(), in.begin()+mid), right_i(in.begin()+mid+1, in.end());
TreeNode* left = reConstructBinaryTree(left_p, left_i);
TreeNode* right = reConstructBinaryTree(right_p, right_i);
TreeNode* root = new TreeNode(pre[0]);
root->left = left;
root->right = right;
return root;
}

size_t findPos(const vector<int>& v, int key) {
for (size_t i = 0; i < v.size(); ++i)
if (v[i] == key)
return i;
return -1;
}
};