LeetCode 105.从前序与中序遍历序列构造二叉树
思路🧐:
由题,前序可以知道根,中序能知道左右子树,那么我们可以进行第一次判断------根为3,左子树为9,右子树为15,20,7。可以利用递归,进行每个结点的判定,比如右子树为15,20,7,而前序顺序为20,15,7,所以可以判断右子树根为20,再一次用中序得出左子树为15,右子树为7,以此类推...
我们可以用rooti(中序数组下标)来切割左右子树 ,rooti-1的范围是左子树,rooti+1的范围是右子树,rooti为需要构建的二叉树结点 ,前序用于找根,中序用于划分,由此反复递归直到整棵树遍历完,就能够构造一颗二叉树了。
代码🔎:
c++class Solution { public: TreeNode* _build(vector<int>& preorder, vector<int>& inorder, int& previ, int inbegin,int inend) { if(inbegin > inend) return nullptr; int rooti = inbegin; //划分下标 while(rooti <= inend) { //当[rooti]==[previ],表示找到根了 if(preorder[previ] == inorder[rooti]) { break; } rooti++; //没找到就++ } //构建结点,并且++前序结点,方便下次rooti划分 TreeNode* root = new TreeNode(preorder[previ++]); //左子树范围在[begin,root-1] root->left = _build(preorder, inorder, previ, inbegin, rooti - 1); //右子树范围在[root+1,end] root->right = _build(preorder, inorder, previ, rooti + 1, inend); return root; } TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) { int i = 0; TreeNode* root = _build(preorder,inorder, i, 0, inorder.size() - 1); return root; } };