二叉树专题--P1030 [NOIP2001 普及组] 求先序排列

[NOIP2001 普及组] 求先序排列

题目描述

给出一棵二叉树的中序与后序排列。求出它的先序排列。（约定树结点用不同的大写字母表示，长度$ \le 8$）。

输入格式

$2$行，均为大写字母组成的字符串，表示一棵二叉树的中序与后序排列。

输出格式

$1$行，表示一棵二叉树的先序。

样例 #1

样例输入 #1

BADC
BDCA

样例输出 #1

ABCD

提示

【题目来源】

NOIP 2001 普及组第三题

#define _CRT_SECURE_NO_WARNINGS 1
#include <bits/stdc++.h>

using namespace std;
//#define int long long
//#define map unordered_map
const int N = 10;
map<char, int> l, r, d;
char in[N], post[N];

int dfs1(int il, int ir, int pl, int pr)
{
	if (pl > pr) return -1;
	char root = post[pr];
	int k = d[root];
	l[root] = dfs1(il, k - 1, pl, k - 1 - il + pl);
	r[root] = dfs1(k + 1, ir, k - il + pl, pr - 1);
	return k;
}

void dfs2(int u)
{
	if (u == -1) return;
	cout << in[u];
	if (l[in[u]] != -1) dfs2(l[in[u]]);
	if (r[in[u]] != -1) dfs2(r[in[u]]);
}

signed main()
{
	int T = 1; //cin >> T;

	while (T--)
	{
		cin >> in >> post;
		for (int i = 0; in[i]; ++i)
		{
			d[in[i]] = i;
		}
		int n = strlen(in);
		int root = dfs1(0, n - 1, 0, n - 1);
		dfs2(root);
		puts("");
	}

	return 0;
}