我们先来看一下题目
二叉树的三个遍历
⑴访问结点本身(N),
⑵遍历该结点的左子树(L),
⑶遍历该结点的右子树(R)。
遍历命名 | 访问规则 |
---|---|
前序遍历 | NLR |
中序遍历 | LNR |
后序遍历 | LRN |
思路解析
对于这个题目,我们知道了二叉树的前序遍历和中序遍历,也就是NLR和LNR的访问循序,那么我们就可以还原二叉树了!
1.首先,在前序遍历的第一个是根节点,我们在中序遍历中可以遍历到这个节点(题目说了树的值是没有重复的,这其实也算是一个提示把!)我们就找到了第一个根节点
2.下面要做的就是找到根节点的左节点和右节点从中序遍历中找到的根节点[下标index]的左边一定是存在根节点的左节点的,又由于是同一棵树,所以前序遍历的根节点的后面index位中也一定存在它的左节点
所以,我们支楞起来了呀! 递归的灵魂出来了呀!
node->left=reBuiled(pre,vin,pre_left+1,pre_left+index-vin_left,vin_left,index-1);
node->right=reBuiled(pre,vin,pre_left+index-vin_left+1,pre_right,index+1,vin_right);
左节点一定在前序遍历的目前节点的右边,而且不会超过之前寻找节点遍历的index-vin_left次数
左节点也一定在中序遍历的目前节点的左边
同理,
右节点一定在前序遍历的目前节点的左边,但一定会在目前节点的index-vin_left+1之后,因为,前序遍历要先遍历完根节点和左节点才会更新右节点
右节点一定在中序遍历的目前节点的右边
代码如下
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* reBuiled(vector<int>& pre,vector<int>& vin,int pre_left,int pre_right,int vin_left,int vin_right){
if(pre_left>pre_right) return nullptr;
int index=0;
for(int i=vin_left;i<=vin_right;i++){
if(vin[i]==pre[pre_left]){
index=i;
break;
}
}
TreeNode* node=new TreeNode(vin[index]);
node->left=reBuiled(pre,vin,pre_left+1,pre_left+index-vin_left,vin_left,index-1);
node->right=reBuiled(pre,vin,pre_left+index-vin_left+1,pre_right,index+1,vin_right);
return node;
}
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
return reBuiled(pre,vin,0,pre.size()-1,0,vin.size()-1);
}
};