由前序遍历和中序遍历重构二叉树

由前序遍历和中序遍历重建二叉树

前序序列(根-左-右):1 2 3 4 5 6

中序序列(左-根-右):3 2 4 1 6 5

1、由前序遍历可知根节点为第一个元素1,在中序遍历序列中找到1对应位置,则1的左边就是左子树3 2 4,右边就是右子树6 5;
2、在找到前序遍历序列中从第二个位置开始往后找相同数目的节点为左子树2 3 4,剩余的就是右子树序列6 5;
3、根据前序遍历规则可知,左子树序列里第一个就是左子树的根节点2,右子树序列里的第一个就是有右子树的根节点5;
4、根据中序遍历序列可知,左子树的根节点2左边的一定是以其为根节点的左子树序列3,右边就是以其为根节点的右子树序列4,右子树的根节点5左边的一定是以其为根节点的右子树序列6

struct BinaryTreeNode
{
int _data;
BinaryTreeNode * _pLeftChild;
BinaryTreeNode* _pRighChild;
};

BinaryTreeNode* RebuildBinaryTree(int* PreOder, int*InOrder, int length)
{
if (PreOder == NULL || InOrder == NULL || length <= 0)
return NULL;
_RebuildBinaryTree(PreOder, PreOder + length - 1, InOrder, InOrder + length - 1);
}
BinaryTreeNode* _RebuildBinaryTree(int* startPre, int* endPre, int* startIn, int* endIn)
{
//根据前序遍历保存根节点
BinaryTreeNode *pRoot;
pRoot = new BinaryTreeNode();
pRoot->_data = startPre[0];
pRoot->_pLeftChild = NULL;
pRoot->_pRighChild = NULL;
//判断是否找完了此次中序遍历,若是找完了,则返回pRoot;
if (startIn == endIn&&*startIn == *endIn)
return pRoot;
//在中序遍历中找根节点所在的位置
int* pRootIn = startIn;
while (*startPre != *pRootIn)
pRootIn++;
//根据根节点在中遍历中的位置,递归还原左子树
int leftlen = pRootIn - startIn;
if (leftlen > 0)
{
pRoot->_pLeftChild = _RebuildBinaryTree(startPre + 1, startPre + leftlen, startIn, startIn + leftLen);
}
//左子树长度加上中序的起始位置后若仍小于整个中序长度,则说明该节点右子树存在,递归还原右子树
if (leftlen + startIn < endIn)
{ 
pRoot->_pRighChild = _RebuildBinaryTree(startPre+leftlen+1,endPre,startIn+leftlen+1,endIn);
}
return pRoot;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,这是一个经典的二叉树重构问题。我们可以通过递归的方式来实现该算法。 首先,我们需要根据前序遍历序列确定该二叉树的根节点。然后,我们可以在中序遍历序列中找到该根节点的位置,以此将整个二叉树划分为左子树和右子树,并且递归处理左右子树。 对于左子树,我们可以利用前序遍历序列和左子树的中序遍历序列递归构建左子树。对于右子树,同样可以利用前序遍历序列和右子树的中序遍历序列递归构建右子树。 最后,我们可以按照后序遍历的顺序输出重构后的二叉树。具体实现可以参考以下代码: ```c++ #include <iostream> #include <vector> #include <unordered_map> using namespace std; struct TreeNode { int val; TreeNode* left; TreeNode* right; TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} }; unordered_map<int, int> index; TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder, int preLeft, int preRight, int inLeft, int inRight) { if (preLeft > preRight) return nullptr; int preRoot = preLeft; int inRoot = index[preorder[preRoot]]; TreeNode* root = new TreeNode(preorder[preRoot]); int leftSubtreeSize = inRoot - inLeft; root->left = buildTree(preorder, inorder, preLeft + 1, preLeft + leftSubtreeSize, inLeft, inRoot - 1); root->right = buildTree(preorder, inorder, preLeft + leftSubtreeSize + 1, preRight, inRoot + 1, inRight); return root; } vector<int> postorderTraversal(TreeNode* root) { vector<int> res; if (!root) return res; vector<int> left = postorderTraversal(root->left); vector<int> right = postorderTraversal(root->right); res.insert(res.end(), left.begin(), left.end()); res.insert(res.end(), right.begin(), right.end()); res.push_back(root->val); return res; } int main() { vector<int> preorder = {3, 9, 20, 15, 7}; vector<int> inorder = {9, 3, 15, 20, 7}; int n = preorder.size(); for (int i = 0; i < n; ++i) { index[inorder[i]] = i; } TreeNode* root = buildTree(preorder, inorder, 0, n - 1, 0, n - 1); vector<int> postorder = postorderTraversal(root); for (int i = 0; i < postorder.size(); ++i) { cout << postorder[i] << " "; } return 0; } ``` 在这个代码中,我们首先利用 unordered_map 存储中序遍历序列中每个节点的位置。然后,我们利用 buildTree 函数递归构建二叉树。最后,我们利用 postorderTraversal 函数按照后序遍历的顺序输出重构后的二叉树。 希望这个代码能够帮助您解决问题。如果您还有其他问题,请随时问我!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值