首页 > 代码库 > 剑指Offer 通过中序和先序遍历重建二叉树
剑指Offer 通过中序和先序遍历重建二叉树
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
写一个reConstruct函数4个形参,分别记录子树前序开始结束的位置,中序开始结束的位子。每次在中序中将前序的根节点找出,讲中序分为前(左子树),后(右子树)2 个部分。递归,直到,子树的开始位置大于结束位置。
AC代码:
1 /** 2 * Definition for binary tree 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8 * }; 9 */10 class Solution {11 public:12 struct TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> in) {13 14 m_pre=pre;15 m_in=in;16 TreeNode* head;17 head=reConstruct(0,pre.size()-1,0,in.size()-1);18 return head;19 }20 21 struct TreeNode* reConstruct(int pre_L,int pre_R,int in_L,int in_R)22 {23 if(pre_L>pre_R||in_L>in_R)24 return nullptr;25 26 TreeNode *root = new TreeNode(m_pre[pre_L]);27 int i=0;28 while(m_pre[pre_L]!=m_in[i])29 i++;30 31 root->left=reConstruct(pre_L+1,pre_L+i-in_L,in_L,i-1);32 root->right=reConstruct(i+pre_L+1-in_L,pre_R,i+1,in_R);33 34 return root;35 }36 37 private:38 //代替全局变量39 vector<int> m_pre;40 vector<int> m_in;41 42 };
剑指Offer 通过中序和先序遍历重建二叉树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。