CF 337D 求圆交

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

题意：就是一棵树上，有一些点被来自东方的神秘力量影响的，力量影响范围是d，为可能的力量源有几个。

思路：相当于是找到距离这m的点的距离都不小于d的点的个数。

先从任意一个点一次dfs，找到m个点中距离最远的点，标记为max1，在以max1开始，dfs一遍，从数组d1记录各个点的距离，找到距离最远的点max2，再从max2开始跑一遍dfs，用d2记录距离。遍历所有点，到max1和max2的距离都小于d的点就成立。

//以上是题解讲的，但是我不懂为什么要先从任意一个点一次dfs，找到m个点中距离最远的点，再从这个点开始DFS。

//我写了一个直接找两个最远的点，求圆交的WA了。等我搞懂了再来补充。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define ll long long
const int maxn=1e5+;
const int INF=0x3f3f3f3f;

int n,m,d;
int a[maxn],d1[maxn],d2[maxn];
int vit[maxn];
int head[maxn],k;

struct Edge
{
    int v;
    int next;
} edge[maxn<<];

void init()
{
    k=;
    memset(head,-,sizeof(head));
    memset(d1,,sizeof(d1));
    memset(d2,,sizeof(d2));
}

void addedge(int u,int v)
{
    edge[k].v=v;
    edge[k].next=head[u];
    head[u]=k++;

    edge[k].v=u;
    edge[k].next=head[v];
    head[v]=k++;
}

void dfs1(int u,int t)
{
    for(int i=head[u]; i!=-; i=edge[i].next)
    {
        int v=edge[i].v;
        if(vit[v]==)
        {
            vit[v]=;
            d1[v]=t+;
            dfs1(v,t+);
        }
    }
}

void dfs2(int u,int t)
{
    for(int i=head[u]; i!=-; i=edge[i].next)
    {
        int v=edge[i].v;
        if(vit[v]==)
        {
            vit[v]=;
            d2[v]=t+;
            dfs2(v,t+);
        }
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d%d%d",&n,&m,&d)==)
    {
        init();
        for(int i=; i<=m; i++) scanf("%d",&a[i]);
        int x,y;
        for(int i=; i<n; i++)
        {
            scanf("%d%d",&x,&y);
            addedge(x,y);
        }

        memset(vit,,sizeof(vit));
        vit[]=;
        dfs2(,);

        int dd=-INF,max1,max2;
        for(int i=; i<=m; i++)
            if(d2[a[i]]>dd) dd=d2[a[i]],max1=a[i];

        memset(vit,,sizeof(vit));
        vit[max1]=;
        dfs1(max1,);

        dd=-INF;
        for(int i=; i<=m; i++)
            if(d1[a[i]]>dd) dd=d1[a[i]],max2=a[i];

        memset(d2,,sizeof(d2));
        memset(vit,,sizeof(vit));
        vit[max2]=;
        dfs2(max2,);

        int ans=;
        for(int i=; i<=n; i++)
            if(d1[i]<=d && d2[i]<=d)
                ans++;
        printf("%d\n",ans);
    }
    return ;
}