PTA-树的遍历

7-8 树的遍历 (25 分)

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式：
输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式：
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。
输入样例：
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
输出样例：
4 1 6 3 5 7 2

---

思路：建树的时候维护一下树的深度，遍历树的时候维护一下同层的访问次序
Code:

#include<bits/stdc++.h>
#define bug(x) cout<<#x<<" = "<<x<<endl;
using namespace std;
typedef long long ll;
const int N = 1e4 + 10;
const int inf = 1e9;
int n;
int a[N],b[N];
struct Node {
	int data;
	int dep;
	Node *left,*right;
};
struct NODE {
	int id;
	int dep;
	int data;
};
Node* build(int x,int y,int z,int w,int dep) {
	if(x>y) return NULL;
	Node *tep;
	tep = new Node;
	int i=z;
	for(; i<=w; i++) if(b[i]==a[y]) break;
	int l1 = i-z;
	tep->data = a[y];
	tep->dep = dep;
	tep->left = build(x,x+l1-1,z,i-1,dep+1);
	tep->right = build(x+l1,y-1,i+1,w,dep+1);
	return tep;
}
vector<NODE> res;
int cnt = 0;
void work(Node *root) {
	if(root) {
		cnt++;
		res.push_back({cnt,root->dep,root->data});
		work(root->left);
		work(root->right);
	}
}
bool cmp(NODE a,NODE b) {
	if(a.dep==b.dep) return a.id<b.id;
	return a.dep < b.dep;
}
int main() {
	cin>>n;
	for(int i=1; i<=n; i++)	cin>>a[i];
	for(int i=1; i<=n; i++)	cin>>b[i];
	Node *root;
	root = new Node;
	root = build(1,n,1,n,1);
	work(root);
	sort(res.begin(),res.end(),cmp);
	for(int i=0; i<res.size(); i++) {
		if(i) cout<<" ";
		cout<<res[i].data;
	}
	return 0;
}

/*
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

*/