(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;
}
}; 
京公网安备 11010502036488号