使用邻接表解决树上问题
#include<bits/stdc++.h>
using namespace std;
int n,d;
int ans;
bool vis[100010]={false};
vector<int> adj[100010];//创建
void dfs(int u,int step)//dfs
{
vis[u]=1;
if(step==d)//对距离特判
{
return;
}
for(int i=0;i<adj[u].size();i++)
{
if(vis[adj[u][i]]!=1)
{
dfs(adj[u][i],step+1);
ans++;//统计数量
}
}
}
int main()
{
cin>>n>>d;
for(int i=1;i<n;i++)
{
int a,b;
cin>>a>>b;//无向图双节点添加
adj[a].push_back(b);
adj[b].push_back(a);
}
dfs(1,0);
cout<<ans;
}

京公网安备 11010502036488号