题目难度:中等
题目考察:二叉树,递归
题目内容:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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); } };