题目难度:中等
题目考察:二叉树,递归
题目内容:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
题目分析
实际上就是给了中序和前序建立二叉树
首先需要知道前中后序的遍历方式:
前序遍历:根左右
中序遍历:左根右
后序遍历:左右根
发现前序遍历的第一个就是根,这就找到了根,可以在中序中找到根,根左边就是左子树,根右边就是右子树
下面继续看先序,2,中序分为了4 7和空,说明4 7在以2为根的左子树,右子树为空
这时候我们发现这实际上是一个递归的过程
这时候就很好写了
递归思路就是先找到到根节点,然后在中序中一分为二,确定左右子树,递归终止条件为子树为空
/**
* 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* helper(vector<int>& pre,vector<int>& vin, int left, int right,int vLeft, int vRight){
if(left>right) return nullptr;
TreeNode* root=new TreeNode(pre[left]);
//根据先序找到当前的根节点
int mid=0;
for(int i=vLeft;i<=vRight;++i){
if(vin[i]==pre[left]){
mid=i;
//在中序中找到根节点确定左右子树
break;
}
}
int leftLen=mid-vLeft;
int rightLen=vRight-mid;
root->left=helper(pre,vin,left+1,left+leftLen,vLeft, mid-1);
//建立左子树
root->right=helper(pre,vin,left+leftLen+1, right, mid+1, vRight);
//建立右子树
return root;
}
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
return helper(pre,vin,0,pre.size()-1,0,vin.size()-1);
}
};
京公网安备 11010502036488号