USACO 3.4 American Heritage

树的遍历转换

最新推荐文章于 2024-03-04 21:13:50 发布

原创最新推荐文章于 2024-03-04 21:13:50 发布 · 382 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树的遍历

本文介绍了一种算法，该算法通过给定的中序和前序遍历序列来构造二叉树，并求解该树的后序遍历序列。利用递归的方法找到每个节点的左右子树，并最终输出后序遍历的结果。

题意：

给出一棵树的中序遍历序列和前序遍历序列，求这棵树的后序遍历序列。

思路：

由前序遍历的第一个元素就是根，在中序遍历划分出左子树和右子树，递归求解。

代码：

/*
ID: jasison2
LANG: C++
TASK: heritage
*/
#include <cstdio>
#include <cstring>
struct Node
{
	int l, r;
	Node(): l(-1), r(-1) {}
	Node(int _l, int _r): l(_l), r(_r) {}
};
Node node[30];
int len, root;
char in[30], pre[30];
void input()
{
	scanf("%s", in);
	scanf("%s", pre);
	len = strlen(in);
	root = pre[0]-'A';
	for (int i = 0; i < 30; ++ i)
		node[i] = Node();
}
void dfs(int il, int ir, int pl, int pr)
{
	if (ir-il <= 0) return;
	int p = il;
	while (in[p] != pre[pl]) p ++;
	if (p != il) node[pre[pl]-'A'].l = pre[pl+1]-'A';
	if (p != ir) node[pre[pl]-'A'].r = pre[pl+p-il+1]-'A';
	dfs(il,p-1,pl+1,pl+p-il);
	dfs(p+1,ir,pl+p-il+1,pr);
}
void output(int x)
{
	if (x != -1)
	{
		output(node[x].l);
		output(node[x].r);
		putchar(x+'A');
	}
}
int main()
{
	freopen("heritage.in","r",stdin);
	freopen("heritage.out","w",stdout);
	input();
	dfs(0,len-1,0,len-1);
	output(root);
	printf("\n");
}