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

给出树的前序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
*/

你可能感兴趣的:(C++树遍历的应用-树的重建)