思路:
我们知道,前序遍历的第一个节点就是树的根节点,所以我们先根据前序遍历序列的第一个数字创建根结点,接下来在中序遍历序列中找到根结点的位置,根节点的左边就是左子树,右边就是右子树,这样就能确定左、右子树结点的数量。在前序遍历和中序遍历的序列中划分了左、右子树结点的值之后,就可以递归地去分别构建它的左右子树。
/** * 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* reConstructBinaryTree(vector<int> pre,vector<int> vin) { if(pre.empty()||vin.empty()||pre.size()!=vin.size()) return nullptr; return reConstructBinaryTreeCore(pre,0,pre.size()-1,vin,0,vin.size()-1); } TreeNode *reConstructBinaryTreeCore(vector<int> pre,int preStart,int preEnd, vector<int> vin,int vinStart,int vinEnd) { if(preStart>preEnd||vinStart>vinEnd) return nullptr; TreeNode *root=new TreeNode(pre[preStart]); for(int i=vinStart;i<=vinEnd;++i) { if(pre[preStart]==vin[i]) { root->left=reConstructBinaryTreeCore(pre,preStart+1,preStart+i-vinStart,vin,vinStart,i-1); root->right=reConstructBinaryTreeCore(pre,preStart+i-vinStart+1,preEnd,vin,i+1,vinEnd); break; } } return root; } };
原文地址:https://www.cnblogs.com/tianzeng/p/11294411.html
时间: 2024-10-11 08:18:51