hdu 1710 Binary Tree Traversals 树的遍历

 

/**//*
    2008-12-28 20:57:43 Accepted 1710 78MS 0K 825 B 
    树的遍历
    给你一棵树的先序遍历结果和中序遍历的结果，让你求以后序遍历输出
    用递归。每次把两个数组分成三个部分，父节点，左子树，右子树，把父节点放到栈里边，重复此步骤直到重建一棵新树
    这时，栈里元素出栈时刚好是后序遍历的顺序
*/
#include<iostream>
#include<stack>
#define N 1001
using namespace std;
stack<int>S;//存放父节点
int pre[N],ino[N];//先序数组和后序数组
int n;
void make(int l1,int r1,int l2,int r2)//l1,r1,是先序遍历的数组的开始和末尾，l2,r2是中序的
{
    int i,j;
    S.push(pre[l1]);    //父节点入栈
    for(i=l2;i<=r2;i++)            //找到父节点在中序遍历的位置i
        if(ino[i]==pre[l1])
            break;
    j=i-l2+l1+1;        //确定左子树和右子树在先序遍历的分界点j，即右子树的父节点
    if(j<=r1 && i+1<=r2)
        make(j,r1,i+1,r2);        //求解右子树
    if(l1+1<=j-1 && l2<=i-1)
        make(l1+1,j-1,l2,i-1);        //求解左子树
}
void print()//输出
{
    while(!S.empty())
    {
        printf("%d",S.top());
        S.pop();
        if(!S.empty())
            printf(" ");
    }
    printf("\n");
}
void init()//输入
{
    for(int i=0;i<n;i++)
        scanf("%d",pre+i);
    for(i=0;i<n;i++)
        scanf("%d",ino+i);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        init();
        make(0,n-1,0,n-1);
        print();
    }
    return 0;
}