(https://uploadfiles.nowcoder.com/images/20200807/72728067_1596814783876_0AE31D4C763C40378CD76AC9C88254C2 "图片标题") 以已知前序遍历和中序遍历为例,其余条件可以类比
基本思路:通过递归分别对左树和右树进行遍历,每次递归都找出根节点的位置。
题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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) {} * }; */
无论是前序还是中序等等,遍历的路径都是一样的,左近右出,绕着树杈爬一圈,只是打印出的根顺序不一样。
对于前序遍历,第一位元素绝逼是整棵树的根,然后是左树再是右树,题目中说不含重复的数字,说明“1”在中序遍历的位置就是根,根左侧为左树,右侧为右树。
C++实现:
第一步:找到中序的根节点位置,只需要遍历两个vector即可
class Solution { public: TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) { //pre为传入的前序结果,vin为传入的中序结果 int len = pre.size(); //如果长度为空,返回null if(len == 0) return NULL; //建立head,根节点为前序遍历第一个数 TreeNode* head = new TreeNode(pre[0]); //找到中序遍历根节点的下标 int rootIndex=0; //3 for(int i = 0; i<len;i++){ if(vin[i] == pre[0]){ rootIndex = i; break; } }
第二步:将当前根左侧和右侧的元素存起来,用于下一次递归再次寻找子树的根
//建立左右元素容器 vector<int> pre_right, pre_left, vin_right, vin_left; //合并左树节点 for(int i=0;i< rootIndex; i++){ pre_left.push_back(pre[i+1]); vin_left.push_back(vin[i]); } //合并右树节点 for(int i=rootIndex+1; i<len; i++){ pre_right.push_back(pre[i]); vin_right.push_back(vin[i]); }
第三步:通过递归逐步建立完整的树。
//递归建立树 //重复寻找(子)树根节点,重新分配左右分支,直到叶,建立完整的树 head->left = reConstructBinaryTree(pre_left, vin_left); head->right = reConstructBinaryTree(pre_right, vin_right); return head; } };