题目描述
A family hierarchy is usually presented by a pedigree tree where all the nodes on the same level belong to the same generation. Your task is to find the generation with the largest population.
输入描述:
Each input file contains one test case. Each case starts with two positive integers N (<100) which is the total number of family members in the tree (and hence assume that all the members are numbered from 01 to N), and M (<N) which is the number of family members who have children. Then M lines follow, each contains the information of a family member in the following format: ID K ID[1] ID[2] ... ID[K] where ID is a two-digit number representing a family member, K (>0) is the number of his/her children, followed by a sequence of two-digit ID's of his/her children. For the sake of simplicity, let us fix the root ID to be 01. All the numbers in a line are separated by a space.
输出描述:
For each test case, print in one line the largest population number and the level of the corresponding generation. It is assumed that such a generation is unique, and the root level is defined to be 1.
输入例子:
23 13 21 1 23 01 4 03 02 04 05 03 3 06 07 08 06 2 12 13 13 1 21 08 2 15 16 02 2 09 10 11 2 19 20 17 1 22 05 1 11 07 1 14 09 1 17 10 1 18
输出例子:
9 4
第一次自己编写的,牛客上排200,也很开心了。用的是层次遍历,额,目前也就这个可以比较号的掌握。这个题目比较清楚也就是二叉树遍历求每一次的节点数,每遍历一层就和上一层比较节点数,保留最大的。估计这个算笨办法。
#include<stdio.h>
#include<iostream>
#pragma warning(disable:4786)
#include<queue>
#include<map>
using namespace std;
map<int,vector<int> > nodes;
int m,n,levelest=0,levelnum=0;//levelest 最大节点数,levelnum 该层数
void levelrange()
{
queue<int> que;
if(nodes[1].size()==0) return;
int par = 1, child ;
int count2=0,count1=1,level=1; //count2 当前层节点数,count1 上一层节点数
int lastLevelMember = 0;
bool first=true;
que.push(par);
while(!que.empty())
{
for(int i=0; i<count1; i++)
{
par= que.front();
que.pop();
for( int j=0; j < nodes[par].size();j++)
{
child = nodes[par][j];
que.push(child);
count2++;
}
}
level++;
lastLevelMember = count1;//上一层节点数,父辈
count1 = count2;
if(levelnum < count2)
{ levelnum=count2;
levelest = level;
}
count2 = 0;
}
}
int main()
{
int pare,cnum,child;
scanf("%d%d",&m,&n);
for(int i=0; i<n; i++)
{
scanf("%d%d",&pare,&cnum);
for(int j=0; j<cnum; j++)
{
cin>>child;
nodes[pare].push_back(child);
}
}
levelrange();//层次遍历
printf("%d %d",levelnum,levelest);;
}
//
