cf 337 D（dfs）

              

                链接：http://codeforces.com/problemset/problem/337/D

  找到已经被攻击的最远距离的两个点，再遍历其他点到这两点之间的距离，当两点之间的距离都小于d，这符合题意  

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int n,m,d;
int a[maxn],dist1[maxn],dist2[maxn],dist3[maxn];
map<int,int>ma;
map<int,int>mark;
vector<int>g[maxn];

void dfs(int v,int cnt,int dist[])
{
    if(mark[v])return;
    mark[v]++;
    dist[v]=cnt;
    for(int i=0;i<g[v].size();i++)
        dfs(g[v][i],cnt+1,dist);
}

int main()
{
    scanf("%d%d%d",&n,&m,&d);
    for(int i=1;i<=m;i++)
    {
        scanf("%d",a+i);
        ma[a[i]]++;
    }
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    mark.clear();
    dfs(1,0,dist1);
    int u=0,v=0,ans=0,mm=0;
    for(int i=1;i<=n;i++)
        if(ma[i]&&mm<dist1[i])
            mm=dist1[i],u=i;
    //cout<<u<<endl;
    mark.clear();
    mm=0;
    dfs(u,0,dist2);
    for(int i=1;i<=n;i++)
        if(ma[i]&&mm<dist2[i])
            mm=dist2[i],v=i;
  //  cout<<v<<endl;
    mark.clear();
    dfs(v,0,dist3);
    for(int i=1;i<=n;i++)
        if(dist2[i]<=d&&dist3[i]<=d)
            ans++;
    printf("%d\n",ans);
    return 0;
}