重建二叉树 来源:牛客网
题目要求:
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
要做这道题需要深刻了解二叉树先序中序的原理,先序遍历指的是先访问此节点,然后访问此节点的左节点,在访问此节点的右节点,而中序是先访问此节点的左节点,再访问该节点,最后访问该节点的右节点。
易知先序遍历的第一个点必为根节点,而中序遍历中左节点必排在父节点之前,因此,可以用递归的方法,不断的找出根节点,代码如下:
class Solution { public: TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) { int inlen=vin.size(); if(inlen==0) return NULL; TreeNode *head=new TreeNode(pre[0]); int rootseri,i; for(i=0;vin[i]!=pre[0];i++); rootseri=i; vector <int > leftpre;vector <int > leftvin; for(i=0;i<rootseri;i++) { leftpre.push_back(pre[i+1]); leftvin.push_back(vin[i]); } vector <int > rightpre;vector <int > rightvin; for(i=rootseri+1;i<inlen;i++) { rightpre.push_back(pre[i]); rightvin.push_back(vin[i]); } head->left=reConstructBinaryTree(leftpre,leftvin); head->right=reConstructBinaryTree(rightpre,rightvin); return head; } };
我超爱学习的