Codeforces-337D Book of Evil【树形dp】

解题思路：

其实就是对于每个点维护它到最远关键点的距离，最后看该距离是否小于d即可。
每个点维护三个值：dis1（到子树中关键点距离的最大值），dis2（到子树中关键点距离的次大值），disup（到不在其子树中关键点距离的最大值）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#define ll long long
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=100005;
int n,m,d,ans;
int tot,first[N],nxt[N<<1],to[N<<1];
int dis1[N],dis2[N],disup[N];
bool key[N];

void add(int x,int y)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y;
}

void dfs1(int u,int fa)
{
    if(key[u])dis1[u]=0;
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];
        if(v==fa)continue;
        dfs1(v,u);
        if(dis1[v]==-1)continue;
        if(dis1[v]+1>dis1[u])dis2[u]=dis1[u],dis1[u]=dis1[v]+1;
        else if(dis1[v]+1>dis2[u])dis2[u]=dis1[v]+1;
    }
}

void dfs2(int u,int fa)
{
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];
        if(v==fa)continue;
        if(dis1[u]==dis1[v]+1)
        {
            if(key[v])disup[v]=max(0,max(dis2[u]+1,disup[u]+1));
            else if(dis2[u]==-1&&disup[u]==-1)disup[v]=-1;
            else if(dis2[u]==-1)disup[v]=disup[u]+1;
            else if(disup[u]==-1)disup[v]=dis2[u]+1;
            else disup[v]=max(dis2[u],disup[u])+1;
        }
        else 
        {
            if(dis1[u]==-1&&disup[u]==-1)disup[v]=-1;
            else if(dis1[u]==-1)disup[v]=disup[u]+1;
            else if(disup[u]==-1)disup[v]=dis1[u]+1;
            else disup[v]=max(dis1[u],disup[u])+1;
        }
        dfs2(v,u);
    }
}

int main()
{
    //freopen("lx.in","r",stdin);
    //freopen("lx.out","w",stdout);
    memset(dis1,-1,sizeof(dis1));
    memset(dis2,-1,sizeof(dis2));
    memset(disup,-1,sizeof(disup));
    int x,y;
    n=getint(),m=getint(),d=getint();
    while(m--)key[getint()]=true;
    for(int i=1;i<n;i++)
    {
        x=getint(),y=getint();
        add(x,y),add(y,x);
    }
    dfs1(1,0);
    dfs2(1,0);
    for(int i=1;i<=n;i++)
    {
        int tmp=max(disup[i],dis1[i]);
        if(tmp==-1)continue;
        else ans+=(tmp<=d?1:0);
    }
    cout<<ans;
    return 0;
}