歡迎使用CSDN-markdown編輯器

L2-006. 樹的遍歷

時間限制
400 ms
內存限制
65536 kB
代碼長度限制
8000 B
判題程序
Standard
作者
陳越
給定一棵二叉樹的後序遍歷和中序遍歷，請你輸出其層序遍歷的序列。這裏假設鍵值都是互不相等的正整數。

輸入格式：

輸入第一行給出一個正整數N（<=30），是二叉樹中結點的個數。第二行給出其後序遍歷序列。第三行給出其中序遍歷序列。數字間以空格分隔。

輸出格式：

在一行中輸出該樹的層序遍歷的序列。數字間以1個空格分隔，行首尾不得有多餘空格。

輸入樣例：
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
輸出樣例：
4 1 6 3 5 7 2

#include<stdio.h>
#include<string.h>

int N;
int t1[50];
int t2[50];
int w1[50];
int w2[50];
int I = 0;
int ans[50][50];
int ansi[50];
int max = 0;
int fun(int d,int *q1,int *q2,int a)
{
//  printf("i%d\n",d);
    if(d == 0) return 0;

    d--;
//  printf("         !!%d!!   **%d\n",q1[d],a);
    ans[a][ansi[a]++] = q1[d];
    int i ;
    for(i = 0 ;i < d;i++)
    {

        if(q2[i] == q1[d])
        {
            break;
        }
    }
    for(int j = 0;j < i;j++)
    {
        w1[j] = q1[j];
//      printf("%d ",w1[j]);
    }
//  printf("\n"); 
    for(int j = 0;j < i;j++)
    {
        w2[j] = q2[j];
//      printf("%d ",w2[j]);
    }
//  printf("\n");
//  printf("!!!%d\n",i);
//  for(int j = 0;j < i;j++)
//  {
//      
//      printf("w1 %d ",w1[j]);
//  }
//      for(int j = 0;j < i;j++)
//  {
//      
//      printf("w2 %d ",w2[j]);
//  }

    fun(i,w1,w2,a+1);   
    int ii = d - i;
//  printf("%d\n",ii); 
    for(int j = 0;j < ii;j++)
    {
        w1[j] = q1[j+i];
//      printf("%d ",w1[j]);
    }
//  printf("\n");
    for(int j = 0;j < ii;j++)
    {
        w2[j] = q2[j+i+1];
//      printf("%d ",w2[j]);
    }

//  printf("\n@\n");

    fun(ii,w1,w2,a+1);
    a>max ? max = a:max = max;
}




int main()
{

    while(~scanf("%d",&N))
    {
        memset(ans,0,sizeof(ans));
        memset(ansi,0,sizeof(ansi));

        for(int i = 0;i < N;i++)
        {
            scanf("%d",&t1[i]); 
        }

        for(int i = 0;i < N;i++)
        {
            scanf("%d",&t2[i]);     
        }
        fun(N,t1,t2,I);
        for(int i = 0 ;i < max;i++)
        {
            for(int j = 0;j < ansi[i];j++)
            {
                printf("%d ",ans[i][j]);
            }
        }
        for(int j = 0;j < ansi[max]-1;j++ )
        {
            printf("%d ",ans[max][j]);
        }
        printf("%d",ans[max][ansi[max]-1]);
    }
 }