C++树遍历的应用-树的重建

最新推荐文章于 2022-06-17 16:57:23 发布

原创最新推荐文章于 2022-06-17 16:57:23 发布 · 258 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #树的遍历

博客介绍了根据树的前序遍历和中序遍历结果，输出后序遍历序列的方法。前序遍历按根—左子树—右子树顺序，中序遍历按左子树—根—右子树顺序。通过前序遍历的当前节点在中序遍历中的位置，可重建左右子树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给出树的前序preorder和中序inorder的结果，输出后序遍历序列。

preorder按照根—左子树—右子树的顺序递归遍历，inorder按照左子树—根—右子树的顺序递归遍历。那么就很容易知道preorder遍历的当前节点c,c在in中的位置m，m左侧就是c的左子树，右侧就是右子树，从而重建以当前c为根的左子树和右子树。

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>

using namespace std;

int n, pos;
vector<int> pre, in, post;

void rec(int l, int r) {
	if (l >= r) return;
	int root = pre[pos++];
	int m = distance(in.begin(), find(in.begin(), in.end(), root));
	rec(l, m);
	rec(m + 1, r);
	post.push_back(root);
}

void solve() {
	rec(0, pre.size());
	for (int i = 0; i < n; i++) {
		if (i) cout << " ";
		cout << post[i];
	}
	cout << endl;
}

int main()
{
	cin >> n;
	int k;
	for (int i = 0; i < n; i++) {
		cin >> k;
		pre.push_back(k);
	}
	for (int i = 0; i < n; i++) {
		cin >> k;
		in.push_back(k);
	}
	solve();
	return 0;
}
/*
input
5
1 2 3 4 5
3 2 4 1 5
*/
/*
output
3 4 2 5 1
*/