PAT 甲级 1004

PAT 甲级 1004
记录下代码

#include<iostream>
#include<stdio.h>
#include<queue>
using namespace std;
int Tree[101][101]={0},depth,nochild[101]={0};
void BFS(int root)
{
    queue<int> Q;
 Q.push(root);
 int lastnode=root,newlastnode;
 depth=1;
 while(!Q.empty())
 {
    int t=Q.front();
    Q.pop();
    int k=0;
    if(Tree[t][k]!=0)
    {
      while(Tree[t][k]!=0)
     {
       Q.push(Tree[t][k]);
      k++;
     }
     newlastnode=Tree[t][k-1];
    }
    else
    {
     nochild[depth]+=1;
    }
    if(t==lastnode)
    {
     lastnode=newlastnode;
     depth++;
    }
 }
}
int main()
{
 int N,M,i,j,k,a,b,num;
 cin>>N>>M;
 if(N==0)
  return 0;
    for(i=0;i<M;i++)
 {
        cin>>a>>num;
  for(j=0;j<num;j++)
  {
   cin>>b;
   Tree[a][j]=b;
  }
 }
 BFS(1);
 for(i=1;i<depth;i++)
 {
  if(i==1)
   cout<<nochild[i];
  else
   cout<<" "<<nochild[i];
 }
 cout<<endl;
 return 0;
}