从中序与后序遍历序列构造二叉树

根据一棵树的中序遍历与后序遍历构造二叉树
假设树中没有重复的元素

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
3
/ \
9 20
… / \
… 15 7

以后序数组的最后一个元素为切割点,先切中序数组,根据中序数组,反过来再切后序数组。一层一层切下去,每次后序数组最后一个元素就是节点元素。

第一步:如果数组大小为0,说明是空节点
第二步:如果不为空,那么取后序数组最后一个元素作为节点元素。
第三步:找到后序数组最后一个元素在中序数组的位置,作为切割点
第四步:切割中序数组,切成中序左数组和中序右数组
第五步:切割后序数组,切成后序左数组和后序右数组
第六步:递归处理左区间和右区间

TreeNode* traversal(vector<int>& inorder, vector<int>& postorder) {
	// 第一步
	if (postorder.size() == 0) return nullptr;
	// 第二步:后序遍历数组最后一个元素,就是当前的中间节点
	int rootValue = postordered[postorder.size() - 1];
	TreeNode* root = new TreeNode(rootValue);
	// 叶子节点
	if (postorder.size() == 1) return root;
	// 第三步:找切割点
	int delimiterIndex;
	for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++)
	{
		if (inorder[delimiterIndex] == rootValue) break;
	}
	// 第四步:切割中序数组,得到中序左数组和中序右数组
	// 第五步:切割后序数组,得到后序左数组和后序右数组
	// 第六步:
	root->left = traversal(中序左数组,后序左数组);
	root->right = traversal(中序右数组,后序右数组);
	return root;
}

首先要切割中序数组
切割点在后序数组的最后一个元素(比如根节点),就是用这个元素来切割中序数组的,所以必要先切割中序数组。
在这里插入图片描述
中序数组切割,找到切割点(后序数组的最后一个元素)在中序数组的位置,然后切割

// 找到中序遍历的切割点
int delimiterIndex;
for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++)
{
	if (inorder[delimiterIndex] == rootValue) break;
}

vector<int> leftInorder(inorder.begin(), inorder.begin() + delimiterIndex);
vector<int> rightInorder(inorder.begin() + delimiterIndex + 1, inorder.end());

然后切割后序数组
首先后序数组的最后一个元素不能要了,这是切割点,也是当前二叉树中间节点的元素,已经用过了。
此时有一个很重的点,就是中序数组大小一定是和后序数组的大小相同的(这是必然)。

中序数组我们都切成了左中序数组和右中序数组了,那么后序数组就可以按照左中序数组的大小来切割,切成左后序数组和右后序数组。

// postorder 舍弃末尾元素,因为这个元素就是中间节点,已经用过了
postorder.resize(postorder.size() - 1);

// 左闭右开,注意这里使用了左中序数组大小作为切割点:[0, leftInorder.size)
vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
// [leftInorder.size(), end)
vector<int> rightPostorder(postorder.begin() + leftInorder.size(), postorder.end());

此时,中序数组切成了左中序数组和右中序数组,后序数组切割成左后序数组和右后序数组。
接下来递归。

class Solution {
private:
    TreeNode* traversal (vector<int>& inorder, vector<int>& postorder) {
        if (postorder.size() == 0) return NULL;

        // 后序遍历数组最后一个元素,就是当前的中间节点
        int rootValue = postorder[postorder.size() - 1];
        TreeNode* root = new TreeNode(rootValue);

        // 叶子节点
        if (postorder.size() == 1) return root;

        // 找到中序遍历的切割点
        int delimiterIndex;
        for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++) {
            if (inorder[delimiterIndex] == rootValue) break;
        }

        // 切割中序数组
        // 左闭右开区间:[0, delimiterIndex)
        vector<int> leftInorder(inorder.begin(), inorder.begin() + delimiterIndex);
        // [delimiterIndex + 1, end)
        vector<int> rightInorder(inorder.begin() + delimiterIndex + 1, inorder.end() );

        // postorder 舍弃末尾元素
        postorder.resize(postorder.size() - 1);

        // 切割后序数组
        // 依然左闭右开,注意这里使用了左中序数组大小作为切割点
        // [0, leftInorder.size)
        vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
        // [leftInorder.size(), end)
        vector<int> rightPostorder(postorder.begin() + leftInorder.size(), postorder.end());

        root->left = traversal(leftInorder, leftPostorder);
        root->right = traversal(rightInorder, rightPostorder);

        return root;
    }
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        if (inorder.size() == 0 || postorder.size() == 0) return NULL;
        return traversal(inorder, postorder);
    }
};
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是实现二叉排序树的代码: #include <stdio.h> #include <stdlib.h> struct Node { int val; struct Node *left; struct Node *right; }; struct Node *insert(struct Node *T, int x) { if (T == NULL) { T = (struct Node *)malloc(sizeof(struct Node)); T->val = x; T->left = NULL; T->right = NULL; } else if (x < T->val) { T->left = insert(T->left, x); } else if (x > T->val) { T->right = insert(T->right, x); } return T; } void postorder(struct Node *T) { if (T == NULL) { return; } postorder(T->left); postorder(T->right); printf("%d ", T->val); } int main() { struct Node *T = NULL; char cmd[10]; int x; while (scanf("%s", cmd) != EOF) { if (cmd[0] == 'i') { scanf("%d", &x); T = insert(T, x); } else if (cmd[0] == 'p') { postorder(T); printf("\n"); } } return 0; } 在这个程序中,我们定义了一个结构体 Node,表示二叉排序树的节点。每个节点包含一个整数值 val,以及指向左右子树的指针 left 和 right。 我们还定义了两个函数 insert 和 postorder,分别用于插入节点和后序遍历二叉排序树。 在主函数中,我们不断读入命令,如果是 insert 命令,则读入一个整数 x 并插入到二叉排序树中;如果是 postorder 命令,则进行后序遍历并输出结果。 注意,我们使用了递归的方式实现插入和后序遍历。在插入节点时,如果当前节点为空,则创建一个新节点并返回;否则,根据节点值的大小递归地插入到左子树或右子树中。在后序遍历时,先递归遍历左子树,再递归遍历右子树,最后输出当前节点的值。 这个程序可以通过以下命令编译运行: gcc -o bst bst.c ./bst 然后输入一系列命令,比如: insert 5 insert 3 insert 7 postorder 输出结果应该为: 3 7 5 这表示后序遍历的结果为 3、7、5。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值