codeforces 337D 树形DP Book of Evil

原题直通车: codeforces 337D Book of Evil

题意：一棵n个结点的树上可能存在一个Evil，Evil危险范围为d，即当某个点与它的距离x<=d时，那么x是危险的。

      现已知道有m个点是危险的，问那个Evil可能存的点有多少。

分析: 昨晚傻X地暴力提交，自然得到的是TLE。今天看一神的代码才突然明白……

      跟以往的题一样，两个DFS就可求出答案。

      第一个DFS搜出所有危险点，并求出枝干上的点到最远的危险点的距离。

      第二次DFS再往回遍历。具体的参考代码。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=111111;
vector<int>Tree[maxn];
int deep[maxn];
int n,m,d,ans;
bool vis[maxn],vid[maxn];
void DFS(int cnt){
    vis[cnt]=true;
    deep[cnt]=-1;
    if(vid[cnt]) deep[cnt]=0;
    int len=Tree[cnt].size();
    for(int i=0;i<len;++i){
        int son=Tree[cnt][i];
        if(vis[son]) continue;
        DFS(son);
        deep[cnt]=max(deep[cnt],deep[son]+(deep[son]==-1?0:1));
    }
}
void DFS_DP(int cnt,int up){ 
    if(up<=d&&deep[cnt]<=d) ans++;
    vis[cnt]=true;
    int len=Tree[cnt].size();
    vector<int>P,S;
    for(int i=0;i<len;++i) {
        int son=Tree[cnt][i];
        if(vis[son]) continue;
        P.push_back(deep[son]==-1?-1:(deep[son]+2));//兄弟结点-->父亲结点-->自身,所以得+2步
        S.push_back(son);
    }
    int ls=S.size();
    if(ls==0) return;
    vector<int>L(ls),R(ls);
    int Max=-1; //下面两循环是为了求出所有兄弟结点的最大deep值。
    for(int i=0;i<ls;++i){
        L[i]=Max;
        if(P[i]>Max) Max=P[i];
    }
    Max=-1;
    for(int i=ls-1;i>=0;--i){
        R[i]=Max;
        if(P[i]>Max) Max=P[i];
    }
    for(int i=0;i<ls;++i){
        int newup=(up==-1?-1:(up+1));
        newup=max(newup,max(L[i],R[i]));
        if(vid[cnt]&&newup<1) newup=1;
        DFS_DP(S[i],newup);
    }
}
int main(){
    cin>>n>>m>>d;
    for(int i=1;i<=m;++i){
        int a; cin>>a; vid[a]=true;
    }
    for(int i=1;i<n;++i) {
        int a,b; cin>>a>>b;
        Tree[a].push_back(b);
        Tree[b].push_back(a);
    }
    DFS(1);
    memset(vis,false,sizeof(vis));
    ans=0;
    DFS_DP(1,-1);
    cout<<ans<<endl;
    return 0;
}