【二叉树】重建,已知前序、中序,已知中序、后序

  对于一棵二叉树而言,遍历方式基本分为三种:前序、中序、后序
  想要通过已有的遍历顺序重建二叉树,需要两种遍历方式,其中必须包括中序遍历。这是因为前序和后序只能提供哪个是根节点(两种方式提供的信息其实是一样的,只不过提供方式稍有不同),不能提供左右子树的信息,想要把左右子树区分开,一定要用中序遍历。因为在中序遍历中,知道了根节点的位置,那么它左边就都是左子树的节点,右边就都是右子树的节点,想要知道根节点的位置,就需要知道根节点是哪个,也就需要前序或者后序遍历的了。

一、已知前序、中序

  在前序遍历中,第一个数,就是根节点,后边的是左子树和右子树,只不过分不出从哪里开始是右子树。每次从前序遍历中取出第一个节点(也就是根节点)后,在中序遍历中找出根节点所在位置,左边的是左子树(这样就知道左子树有多少个节点点了,也就知道前序遍历中,到哪里为止,是左子树了),右边的是右子树,然后递归就可以完成二叉树的构建。

TreeNode* buildTree(vector<int>& pre, vector<int>& in) {
    return helper(pre, 0, pre.size(), in, 0, in.size());
}

TreeNode* helper(const vector<int>& pre, int i, int j, const vector<int>& in, int ii, int jj) {
    // tree        8 4 5 3 7 3
    // preorder    8 [4 3 3 7] [5]
    // inorder     [3 3 4 7] 8 [5]
    // 每次从 pre 头部取一个值 mid, 作为树的根节点
    // 检查 mid 在 in 中的位置, 则 mid 前面部分将作为树的左子树, 右边作为树的右子树
    
    if(i >= j || ii >= j)
        return nullptr;
    const int mid = pre[i];
    const int dis = find(in.begin() + ii, in.begin() + jj, mid) - (in.begin() + ii);
    TreeNode* root = new TreeNode(mid);
    root->left  = helper(pre, i + 1, i + 1 + dis, in, ii, ii + dis);
    root->right = helper(pre, i + 1 + dis, j, in, ii + dis + 1, jj);
    return root;
}

二、已知中序、后序

  同理,后序遍历的最后一个位置一定是根节点,剩下的步骤和上边一样。

TreeNode* helper(const vector<int>& in, int il, int ir, const vector<int>& post, int pl, int pr) {
	if (il >= ir || pl >= pr)
        return nullptr;
    const int mid = post[pr - 1];
	const int dis = find(in.begin() + il, in.begin() + ir, mid) - (in.begin() + il);
	// 左子树有 dis 个节点
	TreeNode* const root = new TreeNode(mid);
	root ->left  = helper(in, il, il + dis, post, pl, pl + dis);
	root ->right = helper(in, il + dis + 1, ir, post, pl + dis, pr - 1);
	return root ;
}

TreeNode* buildTree(vector<int>& in, vector<int>& post) {
	return helper(in, 0, in.size(), post, 0, post.size());
}

三、Notice

  在提交的时候发现一个神奇的事情,就是 TreeNode * root = new TreeNode(mid); 可以过没问题,TreeNode * const root = new TreeNode(mid); 可以过没问题,const TreeNode * root = new TreeNode(mid); 不可以过,因为会提示 root->left 不可修改,也没有问题,但是,const auto root = new TreeNode(mid); 可以过。去测试了一下,发现,const auto a 或者 auto const a 两种方式,都是一个意思,就是 a 是 const,不管 a 是什么类型,是值,就是值不可修改,是指针,就是指针不可修改,而不是指针指向的东西不可修改
  从下边的例子就可以看出来:

  const auto b = new int; 等同于 auto const b = new int; 等同于 int * const b = new int;

四、string 表示的二叉树

  有的题是用 string 表示二叉树,然后输出也是string,比如已知中序 dgbaechf 以及后序遍历 gbdehfca,要你输出前序遍历(adbgcefh),那么其实没什么区别,最后输出的时候是先根,然后左右即可,如下:

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
string process(string in, int il, int ir, string post, int pl, int pr) {
	if (il >= ir || pl >= pr) return string();
	const char root_ch = post[pr - 1];
	const int dis = find(in.begin() + il, in.begin() + ir, root_ch) - (in.begin() + il);
	// 左子树有 dis 个节点
	string left = process(in, il, il + dis, post, pl, pl + dis);
	string right = process(in, il + dis + 1, ir, post, pl + dis, pr - 1);
	string res;
	res += root_ch;
	res += left + right;
	return res;
}

int main() {
	string in = "dgbaechf", post = "gbdehfca";
	cout << process(in, 0, in.size(), post, 0, post.size()) << endl; // 应该 adbgcefh
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值