【北邮OJ】100. 二叉树的层数

【北邮OJ】100. 二叉树的层数

【北邮OJ】100. 二叉树的层数

• • 题目描述
  • 输入格式
  • 输出格式
  • 输入样例
  • 输出样例

题目描述

老师有一个问题想考考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

学不来大佬的sao操作，只能借用队列暴力求解，中规中矩的思路

#include <cstdio>
#include <queue>
using namespace std;
queue <int> q; 
struct node{int count;//记录孩子节点个数 int child[3]; 
}Node[101];int main()
{int t;scanf("%d",&t);int k=1;while(t--){for(int i=0;i<101;i++){Node[i].count=0;}int n,m;scanf("%d%d",&n,&m);for(int i=0;i<n;i++){int a,b;scanf("%d%d",&a,&b);Node[b].child[Node[b].count]=a;Node[b].count++;    		}while(!q.empty()) q.pop();q.push(1);int temp=1;//记录本层所有节点的个数 int num=0;//记录正在遍历到本层的第num个节点 int sum=0;//记录本层子节点的个数，作为下一层所有节点个数使用 printf("Q%d:n",k);k++;while(!q.empty()){int c=q.front();q.pop();printf("%d",c);for(int i=0;i<Node[c].count;i++){int ch=Node[c].child[i];q.push(ch);sum++;}num++;if(num==temp) //本层遍历结束 {printf("n");temp=sum;num=0;sum=0;}else {printf(" ");}}}
}