北邮OJ-100. 二叉树的层数-12网研上机B

以输入边中的子节点为索引来建树(因为子节点才具有唯一性,只会在输入边中提及一次)。先进行一次深度优先搜索在结构体内部记录下该节点的深度,然后进行广度优先搜索打印节点。

题目描述
老师有一个问题想考考mabo,但是mabo不会,所以想请你来帮帮忙。

问题如下:

给一个二叉树

请把这个棵二叉树按层来打印。如果为相同层,需要从左到右打印。一个节点是先添加左节点后添加右节点,即添加顺序与输入顺序一致。

输入格式
首先输入一个整数T,表示一共有T组数据 0< T < =10

再输入两个整数N,M(0<=N,M<=100)

表示下面有N行,这个树有M个节点(1号节点是这棵树的根节点)

每一行两个整数a,b(1<=a,b<=M)

表示节点a的父亲是节点b

输出格式
对于每组

先输出一行 "Qi:"表示第i个问题

然后接下来输出每个问题二叉树每层的节点,在同一层的节点用空格分开,同一层输出在一行(每一行末尾没有空格),不同的层输出在不同行(入下面Sample Ouput所示)

输入样例
2
4 5
2 1
3 1
4 2
5 4
1 2
2 1
输出样例
Q1:
1
2 3
4
5
Q2:
1
2
#include <iostream>
#include <cstdio>
#include <queue>
#define MAXSIZE 500
using namespace std;
typedef struct BiNode{
    int lc;
    int rc;
    int data;
    int deep; 
    bool turnOn;
    BiNode(){//initiate automatically except data 
        initNode();
    }
    void initNode(){
        turnOn=false;
        deep=-1;
        lc=rc=-1;
    }
}*BiTree;
BiNode Tree[MAXSIZE];
int cursor;
int t,n,m;

//非指定下标地申请结点 
//int createNode(int data){//return the index of new node 
// if (cursor+1==MAXSIZE)
// return -1;//failed
// int nowP=cursor;
// Tree[nowP].turnOn=true;
// cursor++;
// Tree[nowP].data=data;
// Tree[nowP].lc=Tree[nowP].rc=-1;
// return nowP;
//}
int createNode(int index){//return the index of new node 
    Tree[index].data=index;
    Tree[index].turnOn=true;
    Tree[index].lc=Tree[index].rc=-1;
    return index;
}

//用来读取输入建树 
void createTree(){
    int a,b;
    createNode(0);
    scanf("%d%d",&n,&m);
    for (int i=0;i<n;i++){
        scanf("%d%d",&a,&b);
        a--;b--;
        createNode(a);
        if (Tree[b].lc==-1)
            Tree[b].lc=a;
        else
            Tree[b].rc=a;
    }
}
void freeNode(int index){
    Tree[index].initNode();
}
void freeTree(int index){//postOrder
    if (index!=-1){
        //traverse
        freeTree(Tree[index].lc);
        freeTree(Tree[index].rc);
        //visit
        Tree[index].initNode();
    }
}

void levelOrder(int root){//root为从0开始的下标 
    queue<BiNode> q;
    int nowDeep=1;
    bool firstCase=true;
    while (!q.empty())
        q.pop();
    q.push(Tree[root]);
    while (!q.empty()){
        BiNode &nowNode=q.front();
        q.pop();
        if (nowNode.deep!=nowDeep){
            printf("\n");
            nowDeep++;
            firstCase=true;
        }
        if (nowNode.lc!=-1)
            q.push(Tree[nowNode.lc]);
        if (nowNode.rc!=-1)
            q.push(Tree[nowNode.rc]);
        //output
        if (firstCase)
            firstCase=false;
        else
            printf(" ");
        printf("%d",nowNode.data+1);
    }
    printf("\n");
}
void calDeep(int root,int nowDeep){//preOrder
    if (root!=-1){
        Tree[root].deep=nowDeep;
        calDeep(Tree[root].lc,nowDeep+1);
        calDeep(Tree[root].rc,nowDeep+1);
    }
}
void preOrder(int root){
    if (root!=-1){
        cout<<Tree[root].deep<<endl;
        preOrder(Tree[root].lc);
        preOrder(Tree[root].rc);
    }
}
int main(){
    scanf("%d",&t);
    for (int time=0;time<t;time++){
        //initiate
        for (int i=0;i<MAXSIZE;i++){
            Tree[i].initNode();
        }
        cursor=0;
        //input
        createTree();
        //caldeep
        calDeep(0,1);
        //output
        printf("Q%d:\n",time+1);
        levelOrder(0);
// preOrder(0);
    }
    return true;
} 

    原文作者:B树
    原文地址: https://blog.csdn.net/u012963208/article/details/64128569
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞