重建二叉树

来源:互联网 发布:mac怎么退出应用 编辑:程序博客网 时间:2024/06/07 20:34
牛客在线编程题题目描述输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。//思路,先序遍历,访问顺序,根节点->左子树->右子树,中序遍历,访问顺序,左子树->根节点->右子树;先序遍历的序列依次取出来的值都为每颗子树的根节点(除第一个点,因为它是整颗树的根节点),结合中序遍历序列,我们可以发现,先序遍历的节点正好将中序遍历的序列分为两部分,左子树和右子树。拿例子来说:
由上面的分析可以知道,结合先序和中序重构二叉树的过程就是不断访问节点的左右子树的递归过程,直到该节点左右子树为空,即该节点为叶子节点,构造该叶子节点,然后往上返回到上一层。这中间主要涉及的就是每次递归时,左右子树的边界设定,知道每次递归的边界,这个问题就解决了。
struct TreeNode {int val;TreeNode* left;TreeNode* right;TreeNode(int x) :val(x), left(NULL), right(NULL){}};TreeNode* reConstructBinaryTree(vector<int> pre, int preStart, int preEnd, vector<int> vin, int vinStart, int vinEnd){if (preStart > preEnd || vinStart > vinEnd){return NULL;}TreeNode* node = new TreeNode(pre[preStart]);for (int i = vinStart; i <= vinEnd; i++){if (pre[preStart] == vin[i]){
//左右子树的边界设定,左子树的边界,从preStart + 1 到preStart + i - vinStart结束
//如果不清楚,可以自己手动试着递归前一两次,就能很好的设定边界。node->left = reConstructBinaryTree(pre, preStart + 1, preStart + i - vinStart, vin, vinStart, i - 1);node->right = reConstructBinaryTree(pre, preStart + i - vinStart + 1, preEnd, vin, i + 1, vinEnd);}}return node;}TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin){TreeNode* root = reConstructBinaryTree(pre, 0, pre.size(), vin, 0, vin.size());return root;}