解题思路
简单使用递归即可。
1、确定根节点,必然是后序遍历的最后一个元素,以此创建树。
2、找到根节点在中序遍历中的位置。
3、递归创建左右子树。
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ #include <algorithm> class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param inorder int整型vector 中序遍历序列 * @param postorder int整型vector 后序遍历序列 * @return TreeNode类 */ TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) { if(inorder.empty()) return nullptr; //返回空树 int root = postorder.back(); //后序遍历的最后一个元素必为根节点 auto tree = new TreeNode(root); //以根节点创建新树 postorder.pop_back(); //后序遍历集合中去除根节点 int pos = find(inorder.begin(), inorder.end(), root) - inorder.begin(); //获取根节点在中序遍历中的位置 vector<int> inLeft(inorder.begin(), inorder.begin() + pos); vector<int> inRight(inorder.begin() + pos + 1, inorder.end()); vector<int> postLeft(postorder.begin(), postorder.begin() + pos); vector<int> postRight(postorder.begin() + pos, postorder.end()); tree->left = buildTree(inLeft, postLeft); tree->right = buildTree(inRight, postRight); return tree; } };