hdu 4707 仓鼠 记录深度 (BFS)

题意：linji的仓鼠丢了，他要找回仓鼠，他在房间0放了一块奶酪，按照抓鼠手册所说，这块奶酪可以吸引距离它D的仓鼠，但是仓鼠还是没有出现，现在给出一张关系图，表示各个房间的关系，相邻房间距离为1，而且图中没有回路，每个房间都是联通的，求仓鼠可能出现的房间的数量。

Sample Input
1
10 2
0 1
0 2
0 3
1 4
1 5
2 6
3 7
4 8
6 9

Sample Output
2

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>

using namespace std;

const int MAXN = 100010;
vector<int>vec[MAXN];
int dep[MAXN];
int pre[MAXN];
void bfs(int s)
{
memset(dep,-1,sizeof(dep));
dep[s] = 0;
queue<int>q;
q.push(s);
while(!q.empty())
{
int u = q.front();
q.pop();
int sz = vec[u].size();
for(int i = 0;i < sz;i++)
{
int v = vec[u][i];

if(dep[v] != -1)continue;
dep[v] = dep[u] + 1;
pre[v] = u;
q.push(v);
}
}
}
int main()
{
//freopen("in.txt","r",stdin) ;
int T;
int n;
int D;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&D);
int u,v;
for(int i = 0;i < n;i++)
vec[i].clear();
for(int i = 1;i < n;i++)
{
scanf("%d%d",&u,&v);
vec[u].push_back(v);
vec[v].push_back(u);
}

bfs(0);
int ans = 0;
for(int i = 0;i < n;i++)
if(dep[i] > D)
ans++;
cout<<ans<<endl;
}
return 0;
}