使用邻接表解决树上问题

#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;
}