Codeforces 337D Book of evil

一道树形dp,写出来是因为最近也做了道类似的.这题是看了分析的思路才做出来的,但感觉很多这样的dp都是利用类似的性质.像这题的话distDown很好想,但distUp的时候就很难想了,其实只要抓住distUp的必然经过父结点或者它的兄弟经过父结点,这周二的多校的那道也是类似的.但是要在线性时间里求出兄弟结点的时候就要注意,我们不可能遍历这个点的所有兄弟结点,所以好的办法就是存最大的两个,当该点是最大的,就用次大的算,其余的都用最大的算.多校的那个也是类似的,不过要存最大,次大,次次大,是有点麻烦.下面贴一记代码初始化为负无穷有点麻烦的样子- -0

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
#define maxn 100000
using namespace std;

int distUp[maxn+];
int distDown[maxn+];
int dp[maxn+][];
int vis[maxn+];
vector<int> G[maxn+];
int n,m,d;

void dfs1(int u)
{
    vis[u]=;
    for(int i=;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(vis[v]) continue;
        vis[v]=;
        dfs1(v);
        if(distDown[v]+>dp[u][]){
            dp[u][]=distDown[v]+;
            if(dp[u][]>dp[u][]){
                swap(dp[u][],dp[u][]);
            }
        }
    }
    distDown[u]=max(distDown[u],dp[u][]);
}

void dfs2(int u)
{
    vis[u]=;
    for(int i=;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(vis[v]) continue;
        if(distDown[v]+==dp[u][]){
            distUp[v]=max(max(dp[u][]+,distUp[v]),distUp[u]+);
        }
        else{
            distUp[v]=max(max(dp[u][]+,distUp[v]),distUp[u]+);
        }
        dfs2(v);
    }
}

void init(int n)
{
    for(int i=;i<=n;i++){
        G[i].clear();
    }
    memset(mark,,sizeof(mark));
    fill(distUp,distUp+n+,-0x3fffffff);
    fill(distDown,distDown+n+,-0x3fffffff);
    for(int i=;i<=n;i++){
        dp[i][]=-0x3fffffff;dp[i][]=-0x3fffffff;
    }
}

int main()
{
    while(cin>>n>>m>>d)
    {
        init(n);
        int tmp;
        for(int i=;i<m;i++){
            scanf("%d",&tmp);
            distDown[tmp]=;
            distUp[tmp]=;
        }
        int tu,tv;
        for(int i=;i<n-;i++){
            scanf("%d%d",&tu,&tv);
            G[tu].push_back(tv);
            G[tv].push_back(tu);
        }
        memset(vis,,sizeof(vis));dfs1();
        memset(vis,,sizeof(vis));dfs2();
        int ans=;
        for(int i=;i<=n;i++){
            if(distUp[i]<=d&&distDown[i]<=d){
                ans++;
            }
        }
        printf("%d\n",ans);
    }
    return ;
}