题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
注:设序列初始长度为n。语言:C++
二叉树结点数据结构规定如下:
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
本题主要采用递归思想,解法如下:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
{
vector<int> pre_lchild, pre_rchild, vin_lchild, vin_rchild;
int i;
int size = pre.size();
if(size == 0)
return NULL;
TreeNode* root = new TreeNode(pre[0]);
for(i = 0; vin[i] != pre[0]; ++i);
pre_lchild = vector<int>(pre.begin()+1, pre.begin()+i+1);
vin_lchild = vector<int>(vin.begin(), vin.begin()+i);
pre_rchild = vector<int>(pre.begin()+i+1, pre.end());
vin_rchild = vector<int>(vin.begin()+i+1, vin.end());
root->left = reConstructBinaryTree(pre_lchild, vin_lchild);
root->right = reConstructBinaryTree(pre_rchild, vin_rchild);
return root;
}
时间复杂度为O(nlogn),空间复杂度为O(n^2)。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。