给一个二叉树的先序和中序遍历,用数组表示,且这两个数组中没有重复的数字。让你重新建立一个二叉树。
其实这个题,我们能看出来这两个遍历之间的关系。先序遍历,数组的第一个就是根节点,第二个就是左子树的根节点(如果有左子树的话),以此类推。中序遍历,树的根节点在数组的中间,左边是左子树右边是右子树。
其实这个规律就是这样的,但是怎么用这个规律来重建一个二叉树呢?
我们想啊,我们开始找的得是整个树的根节点。根据这个根节点我们可以找到,左子树和右子树的根节点。然后依次往下找直到这个节点是个叶子节点就好了。那么我们就可以使用递归来借这个题。
代码:
struct TreeNode
{
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x): val(x),left(NULL), right(NULL) {}
};
class Solution
{
public:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
{
if(pre.size() ==0|| vin.size() == 0)//这两个表示遍历的数组不能为0
return NULL;
TreeNode* root = new TreeNode(pre[0]);//根据先序遍历找到根节点,每次先序遍历的第一个都是这个数的根节点
int i=0;
int len = vin.size();
for(;i<len;++i)
{
if(vin[i] == pre[0])
break;//找到在中序遍历中根节点所在的位置,那么前面的都是左子树后面的是右子树
}
vector<int> pre1(pre.begin()+1,pre.begin()+i+1);
vector<int> pre2(pre.begin()+1+i,pre.end());
vector<int> vin1(vin.begin(),vin.begin()+i);
vector<int> vin2(vin.begin()+i+1,vin.end());
root->left = reConstructBinaryTree(pre1,vin1);
root->right = reConstructBinaryTree(pre2,vin2);
return root;
}
};
网友评论