解题思路
简单使用递归即可。
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;
}
};



京公网安备 11010502036488号