思路

  • 1.把鬼的位置存起来，双向边建树
  • 2.进行一遍dfs找到离树根最远的鬼
  • 3.对该点进行一遍dfs找到离该鬼点最远的鬼点
  • 4.再对鬼点进行一遍dfs，即找出两个相邻最远的鬼点
  • 5.此时如果有点满足到这两个鬼点小于等于d，则成立

代码

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp(aa,bb) make_pair(aa,bb)
#define _for(i,b) for(int i=(0);i<(b);i++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,b,a) for(int i=(b);i>=(a);i--)
#define mst(abc,bca) memset(abc,bca,sizeof abc)
#define X first
#define Y second
#define lowbit(a) (a&(-a))
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
typedef long double ld;
const int N=100010;
const int INF=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-6;
const double PI=acos(-1.0);

int n,m,d,ans;
int p[N];
int dis1[N],dis2[N],dis3[N];
vector<int> g[N];

void dfs1(int u,int fa,int depth){
    dis1[u]=depth;
    for(int v:g[u]){
        if(v==fa) continue;
        dfs1(v,u,depth+1);
    }
}

void dfs2(int u,int fa,int depth){
    dis2[u]=depth;
    for(int v:g[u]){
        if(v==fa) continue;
        dfs2(v,u,depth+1);
    }
}

void dfs3(int u,int fa,int depth){
    dis3[u]=depth;
    for(int v:g[u]){
        if(v==fa) continue;
        dfs3(v,u,depth+1);
    }
}

void solve(){
    cin>>n>>m>>d;
    rep(i,1,m) cin>>p[i];
    rep(i,1,n-1){
        int x,y;cin>>x>>y;
        g[x].pb(y);g[y].pb(x);
    }
    dfs1(1,0,0);
    int temp1=0;
    rep(i,1,m) if(dis1[p[i]]>dis1[temp1]) temp1=p[i];
    dfs2(temp1,0,0);
    int temp2=0;
    rep(i,1,m) if(dis2[p[i]]>dis2[temp2]) temp2=p[i];
    dfs3(temp2,0,0);
    rep(i,1,n) if(dis2[i]<=d&&dis3[i]<=d) ans++;
    cout<<ans<<"\n";
}


int main(){
    ios::sync_with_stdio(0);cin.tie(0);
#ifdef DEBUG
    freopen("F:/laji/1.in", "r", stdin);
//    freopen("F:/laji/2.out", "w", stdout);
#endif
//    int t;cin>>t;while(t--)
    solve();
    return 0;
}