PAT 树的遍历

PAT 树的遍历

• 给定后序和中序遍历，求前序遍历

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
#define debug(a) cout << #a << " " << a << endl
#define x first
#define y second
const int maxn = 1500;
const int N = 1e6 + 7, M = N * 2;
const int inf = 0x3f3f3f3f;
const LL INF = 0xFFFFFFFFFF; //图接近 INT_MAX 时的初始化 判断时使用大于等于
const long long mod = 1e9 + 7;
inline long long read();

// 后序遍历中, 最后一个节点一定是根节点
// 在中序遍历中, 先找到根节点的位置, 左子树在根的左边, 右子树在根的右边
// https://www.acwing.com/solution/content/41325/
// 优化: 找一个节点在中序遍历中是第几个位置出现的 hash表

int n;
int in[N], post[N]; // 存中序遍历序列和后续遍历序列
unordered_map<int, int> l, r; //leftChile[i] = j: i的左孩子是j
unordered_map<int, int> pos; //保存中序遍历中各节点的位置

int build(int il, int ir, int pl, int pr) { // 中序遍历的左右孩子和后序遍历的左右孩子
	int root = post[pr]; // 根节点一定是后序遍历中的最后一个点
	int k = pos[root]; // 根节点在中序遍历中的点
	//画图找
	if(il < k) l[root] = build(il, k - 1, pl, k - il + pl - 1);
	if(ir > k) r[root] = build(k + 1, ir, k - il + pl - 1 + 1, pr - 1);
	return root;
}

void bfs(int root) {
	queue<int> q;
	q.push(root);
	while(q.size()) {
		auto t = q.front();
		q.pop();
		cout << t << ' ';
		if(l.count(t)) q.push(l[t]);
		if(r.count(t)) q.push(r[t]);
	}
}

int main() {

//	freopen("input.txt", "r", stdin);
//	freopen("output.txt", "w", stdout);

//	ios::sync_with_stdio(false);

	int n;
	cin >> n;

	for(int i = 1; i <= n; i++) cin >> post[i];
	for(int i = 1; i <= n; i++) {
		cin >> in[i];
		pos[in[i]] = i;
	}
	int root = build(1, n, 1, n);
	bfs(root);



	return 0;
}


/*
数组开够了吗 开到上界的n+1次方
初始化了吗
*/







inline LL read() {
	char ch = getchar();
	LL p = 1, data = 0;
	while(ch < '0' || ch > '9') {
		if(ch == '-')p = -1;
		ch = getchar();
	}
	while(ch >= '0' && ch <= '9') {
		data = data * 10 + (ch ^ 48);
		ch = getchar();
	}
	return p * data;
}