求二叉树的层次遍历

Bertha 。 2022-07-12 02:54 87阅读 0赞

#### Problem Description ####

已知一颗二叉树的前序遍历和中序遍历，求二叉树的层次遍历。

#### Input ####

输入数据有多组，输入T，代表有T组测试数据。每组数据有两个长度小于50的字符串，第一个字符串为前序遍历，第二个为中序遍历。

#### Output ####

每组输出这颗二叉树的层次遍历。

#### Example Input ####

2
    abc
    bac
    abdec
    dbeac

#### Example Output ####

abc
    abcde

#include<stdio.h>
    #include<string.h>
    #include<stdlib.h>
    #define maxsize 50
    typedef struct node
    {
        char data;
        struct node *lc,*rc;
    }bitree;
    bitree * queue[51];
    int front=0,rear=0;
    bitree * create(int zlen,char qst[],char zst[])
    {
        if(zlen<=0)
            return NULL;
        int i;
        bitree *t;
        t=(bitree *)malloc(sizeof(bitree));
        t->data=qst[0];
        for(i=0;i<zlen;i++)
        {
            if(zst[i]==qst[0])
                break;
        }
        t->lc=create(i,qst+1,zst);
        t->rc=create(zlen-i-1,qst+i+1,zst+i+1);
        return t;
    }
    void enter_queue(bitree *t)
    {
        if((rear+1)%maxsize!=front)
        {
            rear=(rear+1)%maxsize;
            queue[rear]=t;
        }
    }
    bitree *delete_queue()
    {
        if(rear!=front)
        {
            front=(front+1)%maxsize;
            return queue[front];
        }
    }
    void level_order(bitree *t)
    {
        bitree *p;
        if(t)
        {
            enter_queue(t);
            while(rear!=front)
            {
                p=delete_queue();
                printf("%c",p->data);
                if(p->lc)
                {
                    enter_queue(p->lc);
                }
                if(p->rc)
                {
                    enter_queue(p->rc);
                }
            }
        }
    }
    int main()
    {
        int zlen,t;
        char qst[51],zst[51];
        bitree * tree;
        scanf("%d",&t);
        while(t--)
        {
            scanf("%s%s",qst,zst);
            zlen=strlen(zst);
            tree=create(zlen,qst,zst);
            level_order(tree);
            printf("\n");
        }
        return 0;
    }