其他分享
首页 > 其他分享> > PATA 1020 Tree Traversals

PATA 1020 Tree Traversals

作者:互联网

题目大意:
给定二叉树的后根序列和中根序列,输出层次序列。

输入:
第一行:节点个数
第二行:后根序列
第三行:中根序列
输出:
层次序列,数字间用一个空格隔开,末尾不允许有多余空格。

代码:

#include <stdio.h>
#include <queue>

using namespace std;

const int maxn=40;

int post[maxn],in[maxn];
int N;

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

biTreeNode* create(int postL,int postR,int inL,int inR){
    if(postL>postR){
        return NULL;//该结点为空指针,即没有结点
    }
    biTreeNode * root=new biTreeNode;//定义结构体指针变量需要new
    root->data=post[postR];
    int i;
    for(i=inL;i<=inR;i++){
        if(in[i]==post[postR]){
            break;
        }
    }
    //这里一定要小心,一定需要算出来左子树的结点个数,因为后根序列和中根的坐标后面是错开的
    int k=i-inL;
    root->lchild=create(postL,postL+k-1,inL,inL+k-1);
    root->rchild=create(postL+k,postR-1,inL+k+1,inR);
    return root;
}

int num=0;//计算输出的结点个数,用来控制空格输出,使得最后一个数字后没有空格
void level(biTreeNode * root){
    queue<biTreeNode*> q;
    q.push(root);
    while(!q.empty()){
        biTreeNode* 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 j=0;j<N;j++){
        scanf("%d",&in[j]);
    }
    biTreeNode* root=create(0,N-1,0,N-1);
    level(root);
    return 0;
}

标签:inL,int,Traversals,Tree,biTreeNode,postR,PATA,postL,root
来源: https://blog.csdn.net/abracadabraa/article/details/113921084