九度OJ - 1078 - 二叉树遍历

题目描述

二叉树的前序、中序、后序遍历的定义：
前序遍历：对任一子树，先访问跟，然后遍历其左子树，最后遍历其右子树；
中序遍历：对任一子树，先遍历其左子树，然后访问根，最后遍历其右子树；
后序遍历：对任一子树，先遍历其左子树，然后遍历其右子树，最后访问根。
给定一棵二叉树的前序遍历和中序遍历，求其后序遍历（提示：给定前序遍历与中序遍历能够唯一确定后序遍历）。

输入

两个字符串，其长度n均小于等于26。
第一行为前序遍历，第二行为中序遍历。
二叉树中的结点名称以大写字母表示：A，B，C…最多26个结点。

输出

输入样例可能有多组，对于每组测试样例，
输出一行，为后序遍历的字符串。

样例输入

ABC
BAC
FDXEAG
XDEFAG

样例输出

BCA
XEDGAF

来源

2006年清华大学计算机研究生机试真题

先序遍历：

• 访问根节点
• 先序遍历左子树
• 先序遍历右子树

中序遍历：

• 中序遍历左子树
• 访问根节点
• 中序遍历右子树

后序遍历：

• 后序遍历左子树
• 后序遍历右子树
• 访问根节点

因为先序遍历的第一个节点一定是根，所以从先序遍历着手，根据递归，先遍历左子树，再遍历右子树

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int MAXN = 30;
char preOrder[MAXN];    //前序 
char inOrder[MAXN];     //中序 
char ans[MAXN];
int cnt;


//// 返回root在前序中的位置
int GetPreOrderLoc(char root){
	int len = strlen(preOrder);
	for(int i = 0; i < len; i++){
		if(preOrder[i] == root){
			return i;
		}
	}
}

// 返回root在中序中的位置
int GetInOrderLoc(char root){
	int len = strlen(inOrder);
	for(int i = 0; i < len; i++){
		if(inOrder[i] == root){
			return i;
		}
	}
}

//递归处理子树preOrder[l...r]
//root为该子树的根节点，l r指的是在中序中的位置 
void dfs(char root, int l, int r){
	if(l <= r){
		int p1 = GetPreOrderLoc(root);    //root在前序中的位置
		int p2 = GetInOrderLoc(root);     //root在中序中的位置 
		//递归左子树
		dfs(preOrder[p1+1], l, p2-1);
		//递归右子树 
		dfs(preOrder[p1+p2-l+1], p2+1, r);
		ans[cnt++] = root;
	}
}

int main()
{
	while(scanf("%s %s", preOrder, inOrder)!=EOF){
		cnt = 0;
		dfs(preOrder[0], 0, strlen(preOrder)-1);
		
		for(int i = 0; i < cnt; i++){
			cout << ans[i];
		}
		cout << endl;
	} 
	return 0;
}