问题:从前序与中序遍历序列构造二叉树
解题思路
前序遍历:根、左、右
中序遍历:左、根、右
前序遍历向量中的第一个元素就是根节点,我们通过根节点就可以将中序遍历分割成3部分,左、根、右,递归建立即可
C++代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
//如果没节点了返回NULL
if(preorder.size() == 0 || inorder.size() == 0) return NULL;
vector<int>left, right;
//将inorder中在根节点pre[0]左边的数归入left中
int i = 0, size = inorder.size();
while(i < size && inorder[i] != preorder[0]){
left.push_back(inorder[i]);
i++;
}
//将inorder中在根节点pre[0]右边的数归入right中
i++;
while(i < size){
right.push_back(inorder[i]);
i++;
}
TreeNode *root = new TreeNode(preorder[0]);//建立根节点
preorder.erase(preorder.begin());//删除第一个元素
root->left = buildTree(preorder, left);//递归建立左子树
root->right = buildTree(preorder, right);//递归建立右子树
return root;//返回根节点
}
};
问题:从中序与后序遍历序列构造二叉树
解题思路
1、通过后序遍历的最后一个节点获得根节点的值
2、根据根节点值将中序遍历序列进行分割,左右子树分别存放在left和right中
3、递归建立右子树,然后递归建立左子树
C++代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
//如果没节点了返回NULL
if(inorder.size() == 0 || postorder.size() == 0) return NULL;
//根据根节点值将中序遍历序列进行分割,左右子树分别存放在left和right中
vector<int>left, right;
int rootVal = postorder.back();//根节点值
//将inorder中在根节点rootVal左边的数归入left中
int i = 0, size = inorder.size();
while(i < size && inorder[i] != rootVal){
left.push_back(inorder[i]);
i++;
}
//将inorder中在根节点rootVal右边的数归入right中
i++;
while(i < size){
right.push_back(inorder[i]);
i++;
}
postorder.pop_back();//在后序遍历序列中删除根节点
TreeNode *root = new TreeNode(rootVal);//建立根节点
root->right = buildTree(right, postorder);//注意:先递归建立右子树
root->left = buildTree(left, postorder);//递归建立左子树
return root;//返回根节点
}
};