CodeForces337D(树形DP）

新手还是不要看我这一篇博客，今天没时间解释，这个是给我自己写的，代码没有经过整理，很烂！

大概题意：给一棵树（n个点，边权为1），然后告诉你树上某个点有个污染物，那个污染物会影响距离污染物不超过d的点，然后告诉你m个已确定被影响的点（其他点不确定，就是可能被影响，可能没影响），问有多少个点可能是那个污染物所在的点。样例1:点1,2确定被影响，那么污染物可能在3,4或者5（3种可能），所以答案输出3。

树形DP的一个经典模型：

第一个dfs只考虑的后代的影响，第一次dfs后，因为根节点已经有了最后的答案，所以第二次dfs从根节点开始，往下更新，加入祖先的影响。代码：

#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
using namespace std;

const int N=100010;
int id[N],id2[N],dis[N],dis2[N],vis[N];
vector<int> vec[N];

void dfs1(int u,int f){
    for(int i=0;i<vec[u].size();i++){
        int v=vec[u][i];
        if(v==f) continue;
        dfs1(v,u);
        if(dis[v]!=-1&&dis[v]+1>dis2[u]){
            dis2[u]=dis[v]+1;
            id2[u]=v;
        }
        if(dis2[u]>dis[u]){
            swap(dis2[u],dis[u]);
            swap(id[u],id2[u]);
        }
    }
}

void dfs2(int u,int f){
    if(f!=-1){
        if(id[f]==u){
            if(dis2[f]!=-1&&dis2[f]+1>dis2[u]){
                dis2[u]=dis2[f]+1;
                id2[u]=f;
            }
            if(dis2[u]>dis[u]){
                swap(dis[u],dis2[u]);
                swap(id[u],id2[u]);
            }
        }else{
            if(dis[f]!=-1&&dis[f]+1>dis2[u]){
                dis2[u]=dis[f]+1;
                id2[u]=f;
            }
            if(dis2[u]>dis[u]){
                swap(dis[u],dis2[u]);
                swap(id[u],id2[u]);
            }
        }
    }
    for(int i=0;i<vec[u].size();i++){
        int v=vec[u][i];
        if(v==f) continue;
        dfs2(v,u);
    }
}

int main(){
    int n,m,d,u,v;
    while(scanf("%d%d%d",&n,&m,&d)==3){
        for(int i=1;i<=n;i++) vec[i].clear();
        memset(id,-1,sizeof(id));
        memset(id2,-1,sizeof(id2));
        memset(dis,-1,sizeof(dis));
        memset(dis2,-1,sizeof(dis2));

        for(int i=1;i<=m;i++){
            scanf("%d",&u);
            dis[u]=0;
        }
        for(int i=1;i<=n-1;i++){
            scanf("%d%d",&u,&v);
            vec[u].push_back(v);
            vec[v].push_back(u);
        }

        dfs1(1,-1);
        dfs2(1,-1);
        int ans=0;
        for(int i=1;i<=n;i++){
            if(dis[i]<=d)
                ans++;
        }
        printf("%d\n",ans);
    }
    return 0;
}