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

断了今生、忘了曾经 提交于 2020-01-29 01:35:58

输入:
    第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();
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!