CodeForces Round #196 D Book of Evil

#include<bits/stdc++.h>
using namespace std;

const int maxn = 112345;

vector<int> edge[maxn];

void init(int n){
    for(int i=0;i<=n;i++)
        edge[i].resize(0);
}

void Link(int st,int ed){
    edge[st].push_back(ed);
    edge[ed].push_back(st);
}

bool newt[maxn];
bool vis[maxn];

bool dfsg(int st){
    vis[st]=true;
    for(auto &x:edge[st]){
        if(vis[x]==false){
            vis[x]=true;
            if(dfsg(x)){
                newt[st]=true;
            }
        }
    }
    return newt[st];
}

int dis[maxn];

void dfsnd(int st,int d){
    dis[st]=d;
    for(auto &x:edge[st]){
        if(newt[x]==true && dis[x]==-1){
            dfsnd(x,d+1);
        }
    }
}

int disa[maxn];
int disb[maxn];

void dfs(int st,int d,int *dis){
    dis[st]=d;
    for(auto & x : edge[st]){
        if(dis[x]==-1)
            dfs(x,d+1,dis);
    }
}

int main(){
    int n,m,d;
    while(~scanf("%d %d %d",&n,&m,&d)){
        memset(newt,0,sizeof(newt));
        init(n);
        int st;
        while(m--){
            scanf("%d",&st);
            newt[st]=true;
        }
        int u,v;
        for(int i=1;i<n;i++){
            scanf("%d %d",&u,&v);
            Link(u,v);
        }

        memset(vis,0,sizeof(vis));
        dfsg(st);

        memset(dis,-1,sizeof(dis));
        dfsnd(st,0);
        int sta = st;
        for(int i=1;i<=n;i++){
            if(dis[i] > dis[sta])
                sta = i;
        }

        memset(dis,-1,sizeof(dis));
        dfsnd(sta,0);

        int stb = sta;
        for(int i=1;i<=n;i++){
            if(dis[i] > dis[stb])
                stb = i;
        }


        memset(disa,-1,sizeof(disa));
        memset(disb,-1,sizeof(disb));

        dfs(sta,0,disa);
        dfs(stb,0,disb);

        int ans=0;

        for(int i=1;i<=n;i++){
            if(max(disa[i],disb[i]) <=d)
                ans++;
        }
        printf("%d
",ans);
    }
    return 0;
}