Problem C: 二叉树遍历
题目
Description
二叉树的前序、中序、后序遍历的定义:
前序遍历:对任一子树,先访问跟,然后遍历其左子树,最后遍历其右子树;
中序遍历:对任一子树,先遍历其左子树,然后访问根,最后遍历其右子树;
后序遍历:对任一子树,先遍历其左子树,然后遍历其右子树,最后访问根。
给定一棵二叉树的前序遍历和中序遍历,求其后序遍历(提示:给定前序遍历与中序遍历能够唯一确定后序遍历)。
Input
两个字符串,其长度n均小于等于26。
第一行为前序遍历,第二行为中序遍历。
二叉树中的结点名称以大写字母表示:A,B,C…最多26个结点。
Output
输入样例可能有多组,对于每组测试样例,
输出一行,为后序遍历的字符串。
Sample Input Copy
ABC
CBA
ABCDEFG
DCBAEFG
Sample Output Copy
CBA
DCBGFEA
思路
- 先由先序遍历和中序遍历推导二叉树,存入结点
- 该题重点在如何判断根结点和左右子树
- 每个子树的根结点都在先序遍历的最前端,由此可以找到根结点保存
- 每个子树的左子树都在中序遍历中根结点左侧,所以可以通过查找到中序中根结点所在位置找到左子树(注意其结点数量与先序遍历相同)
- 每个子树的左子树都在中序遍历中根结点右侧
- 因此可以得到该结点左子树和右子树的先序和中序序列,得以进行递归
- 再通过后序遍历(左右中顺序,使用递归(dfs))输出
- 如下图(摘选自算法笔记)
代码
#include <stdio.h>
#include <iostream>
using namespace std;
struct Node{
char num;
Node* left=NULL;
Node* right=NULL;
};
string pre;
string in;
Node* resetTree(int preL,int preR,int inL,int inR){//注意不能把Node作为参数,否则会为null
if(preL>preR||inL>inR)//注意判断是否这一侧子树已经不存在结点
return NULL;
Node* root=new Node;
root->num=pre[preL];
if(preL==preR)
return root;
int i=inL;
for(i=inL;i<=inR;i++){
if(pre[preL]==in[i])
break;
}
root->left=resetTree(preL+1,preL+i-inL,inL,i-1);//左子树
root->right=resetTree(preL+i-inL+1,preR,i+1,inR);//右子树
return root;
}
void postorder(Node* root){
if(root==NULL)
return;
postorder(root->left);
postorder(root->right);
printf("%c",root->num);
return;
}
int main(){
while(getline(cin,pre)){
getline(cin,in);
Node* root=resetTree(0,pre.length()-1,0,in.length()-1);
postorder(root);
printf("\n");
}
return 0;
}