1076. Forwards on Weibo (30)

所有的测试点通过，除了最后一个运行超时

#include<iostream>
using namespace std;
#define max 10005
int map[max][max];
int visited[max];
int level[max];
int n,l;//用户数和深度


void dfs(int id)
{
visited[id]=1;
for(int i=1;i<=n;i++)
{
if(map[i][id]&&level[i]>level[id]+1)
{
level[i]=level[id]+1;
visited[i]=0;
}
if(map[i][id]&&!visited[i])
{

level[i]=level[id]+1;
visited[i]=1;
dfs(i);
}
}
}
int main()
{
int i,j,m,list,k,id,count;
cin>>n>>l;
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
map[i][j]=0;
}
}
for(i=1;i<=n;i++)
{
cin>>m;
for(j=0;j<m;j++)
{
cin>>list;
map[i][list]=1;
}
}

/* for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
cout<<"map["<<i<<"]"<<"["<<j<<"]="<<map[i][j]<<endl;
}
}*/


cin>>k;
while(k--)
{
count=0;
for(i=1;i<=n;i++)
{
level[i]=0;
visited[i]=0;
}
cin>>id;
dfs(id);
for(i=1;i<=n;i++)
{

if(level[i]>0&&level[i]<=l)count++;
}


cout<<count<<endl;
}
system("pause");
return 0;
}