二叉树遍历之已知前序中序求后序

Problem Description

 

题目1078：二叉树遍历

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：1529

解决：934

题目描述：

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

输入：

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

输出：

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

样例输入：

ABC
BAC
FDXEAG
XDEFAG

样例输出：

BCA
XEDGAF

来源：

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

Problem Solutions

思路

由前序遍历可以知道二叉树的根节点

由中序遍历可以将根节点左边划分为左子树,根节点右边划分为右子树

知道了左子树和右子树各自的节点个数之后就可以得到左子树的前序遍历和中序遍历结果以及右子树的前序遍历和中序遍历结果.

这样就就可以将问题规模缩小从而递归解决问题.

在求解过程中需要注意处理子树只有一个节点的情形和子树为空的情形.

比如说前序遍历结果为AG,中序遍历结果为AG,那么左子树为空,这种情况要注意.

MYCode

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define MAX 30
char str1[MAX];
char str2[MAX];
struct node
{
    int left;
    int right;
};
node tree[MAX];
void init()
{
    int i;
    for(i=0;i<MAX;i++)
    {
        tree[i].left=-1;
        tree[i].right=-1;
    }
}
void build(char str1[MAX],char str2[MAX])
{
    int len=strlen(str1);
    if(len==1||len==0)//叶子节点或者空节点
    return;
    //cout<<"build "<<str1<<" "<<str2<<endl;
    int root_index=str1[0]-'A';
    int left_index=str1[1]-'A';
    int i,j;
    int pos=0;
    for(i=0;i<len;i++)
    {
        //二叉树中所有节点都是字母并且各不相同
        if(str2[i]==str1[0])
        {
            pos=i;
            break;
        }
    }
    char left_prefix[MAX],left_mid[MAX],right_prefix[MAX],right_mid[MAX];

    int left_len=pos;
    int right_len=len-left_len-1;
    
    //获得右子树的根节点
    int right_index=str1[left_len+1]-'A';
    if(left_len!=0)//左子树是否为空
    tree[root_index].left=left_index;
    if(right_len!=0)//右子树是否为空
    tree[root_index].right=right_index;
    
    
    for(i=1,j=0;i<=left_len;i++)
    left_prefix[j++]=str1[i];
    for(i=0,j=0;i<left_len;i++)
    left_mid[j++]=str2[i];
    left_prefix[j]='\0';
    left_mid[j]='\0';

    for(i=left_len+1,j=0;i<len;i++)
    right_prefix[j++]=str1[i];
    for(i=left_len+1,j=0;i<len;i++)
    right_mid[j++]=str2[i];
    right_prefix[j]='\0';
    right_mid[j]='\0';

    //cout<<left_prefix<<" "<<left_mid<<" "<<right_prefix<<" "<<right_mid<<endl;
    build(left_prefix,left_mid);
    build(right_prefix,right_mid);
}
void suffix(int root)
{
    if(tree[root].left!=-1)
    suffix(tree[root].left);
    if(tree[root].right!=-1)
    suffix(tree[root].right);
    cout<<(char)('A'+root);
}
int main()
{
    while(scanf("%s%s",&str1,&str2)!=EOF)
    {
        init();
        int root_index=str1[0]-'A';
        build(str1,str2);
        suffix(root_index);
        cout<<endl;
    }
}

解法和代码都不够简洁.如果有更简洁的解法和代码欢迎留言.

测试提交口

http://ac.jobdu.com/problem.php?pid=1078