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

输入:
    第1行输入二叉树的结点数
    第2行输入前序遍历的结点编号序列,相邻编号用空格分开
    第3行输入中序遍历的结点编号序列,相邻编号用空格分开

输出:
    在1行中输出后序遍历的结点编号序列,用空格分开

代码

#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
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(){
	pos = 0;
	rec(0, pre.size());
	for(int i = 0; i < n; i++){
		if(i) cout << " ";
		cout << post[i];
	}
	cout << endl;
}

int main(){
	int k;
	cin >> n;
	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();
}
发布了54 篇原创文章 · 获赞 17 · 访问量 9160

猜你喜欢

转载自blog.csdn.net/qq_41979513/article/details/103704830