2013年8月25日 星期日

2013/8/25 Codeforces 337D Book of Evil

// http://codeforces.com/contest/337/problem/D
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
int s[100001][2]; int L1[100001], L2[100001]; bool visit[100001];
vector<int> v[100001];
int dfs1(int u)
{
    visit[u] = 1;

    for(int i = 0; i < v[u].size(); i++)
    {
        int t = v[u][i];
        if( visit[t] == 1 ) continue;

        int y = dfs1(t);
        if( y == -1 ) continue;

        if( y+1 > s[u][0] ){ s[u][1] = s[u][0]; s[u][0] = y+1; }
        else if( y+1 > s[u][1] ){ s[u][1] = y+1; }
    }

    L1[u] = max(L1[u], s[u][0]);

    return L1[u];
}
void dfs2(int u)
{
    visit[u] = 1;

    for(int i = 0; i < v[u].size(); i++)
    {
        int t = v[u][i];
        if( visit[t] ) continue;

        if( L2[u] != -1 ) L2[t] = L2[u] + 1;

        if( L1[t]+1 == s[u][0] )
        {
            if( s[u][1] != -1 ) L2[t] = max(L2[t], s[u][1]+1);
        }
        else if( s[u][0] != -1 )
            L2[t] = max(L2[t], s[u][0]+1);

        dfs2(t);
    }

    return;
}

int main()
{
    int n, m, d;
    scanf("%d %d %d", &n, &m, &d);

    for(int i = 1; i <= n; i++)
        s[i][0] = s[i][1] = L1[i] = L2[i] = -1;

    for(int i = 0; i < m; i++)
    {
        int t; scanf("%d", &t);
        L1[t] = L2[t] = s[t][0] = 0;
    }

    for(int i = 1; i < n; i++)
    {
        int a, b; scanf("%d %d", &a, &b);
        v[a].push_back(b); v[b].push_back(a);
    }

    for(int i = 1; i <= n; i++)
        visit[i] = 0;
    dfs1(1);

    for(int i = 1; i <= n; i++)
        visit[i] = 0;
    dfs2(1);

    int ans = 0;
    for(int i = 1; i <= n; i++)
        if( max(L1[i], L2[i]) <= d )
            ans++;

    printf("%d\n", ans);

}

沒有留言:

張貼留言