已知先、中序,求层序输出------PAT_A1020 Tree Traversals (25 分)

1020 Tree Traversals (25 分)
假设二叉树中的所有值都是不同的正整数。给定后序和中序遍历序列,输出相应二叉树的层序遍历序列。
输入规范
每个输入文件包含一个测试用例。对于每种情况,第一行给出一个正整数N(≤30),即二叉树中的节点总数。第二行给出后序,第三行给出了中序。一行中的所有数字都用空格隔开。
输出规范
对于每个测试用例,在一行中打印相应二叉树的层序遍历序列。一行中的所有数字必须用一个空格隔开,并且在行尾不能有多余的空格。

思路:与已知先序序列、中序序列,求二叉树一样。

代码

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;

const int maxn = 50;

struct node {
    int data;
    node* lchild;
    node* rchild;
};

int in[maxn], post[maxn];
int n;                    //结点数
//先由已知求出树
node* creat ( int postL, int postR, int inL, int inR ){
    if( postL > postR ){
        return NULL;
    }
    node* root = new node;
    root->data = post[postR];
    int k;
    for ( k=inL; k<=inR; k++ ){
        if( in[k] == post[postR] ){
            break;
        }
    }
    int numLeft = k-inL;
    root->lchild = creat ( postL, postL+numLeft-1, inL, k-1 );
    root->rchild = creat ( postL+numLeft, postR-1, k+1, inR );
    
    return root;
}
int num = 0;                     //记录已输出结点个数
//层序遍历输出
void BFS ( node* root ){
    queue <node*> q;
    q.push(root);
    while ( !q.empty() ){
        node* now = q.front();
        q.pop();
        printf ( "%d", now->data );
        num++;
        if( num < n ){
            printf(" ");
        }
        if ( now->lchild != NULL ) q.push ( now->lchild );
        if ( now->rchild != NULL ) q.push ( now->rchild );
    }
}

int main ( )
{
    scanf("%d", &n);
    for ( int i=0; i<n; i++ ){
        scanf("%d", &post[i]);
    }
    for ( int i=0; i<n; i++ ){
        scanf("%d", &in[i]);
    }
    node* root = creat ( 0, n-1, 0, n-1 );
    BFS ( root );
    
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值