题目链接:http://codeforces.com/problemset/problem/337/D
参考博客:http://www.cnblogs.com/chanme/p/3265913
题目大意:给你一个n个点的无向树。任意两点的距离为中间经过的边数。现在某个点上有本魔法书,这本书对与这个点距离小于等于d的点有影响。给你收集到的m个受影响的点(信息不一定全对)。要你判断有多少个点可能放魔法书。
算法思路:我参考别人的想法,自己开始怎么也想不出好的算法,n也太大。这题用树形dp,两遍dfs来统计出每一个点到所有这个m个受影响点的距离的最大值。只要这个最大值<=d,就可能放魔法书。 所以关键就是算这个最大值,有树形dp第一遍就可以统计以u为根,u到子树中存在的受影响点的最大值。第二遍要利用父亲和兄弟节点来求出u到剩余受影响点(即不在u的子树上的点)的最大值。然后和在一起就是u到所有受影响点的最大值。
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std; const int maxn = ;
const int INF = 0x3f3f3f3f; int disDown[maxn]; //disDown[u]表示以u为根的子树Pm中的点到u距离的最大值。
int disUp[maxn]; //disUp[u]表示以u为根去掉上面的子树中的点,u到与父亲相连的所有pm的最大值。
int Max[maxn]; //Max[u]表示u为根子树Pm中的点到u距离的最大值。
int SecMax[maxn]; //SecMax[u]表示u为根子树Pm中的点到u距离的第二大值。
int n,m,d;
vector<int> G[maxn]; void dfs1(int u,int fa)
{
for(int i=,sz=G[u].size(); i<sz; i++)
{
int v = G[u][i];
if(v == fa) continue; dfs1(v,u);
if(disDown[v]+ > SecMax[u])
{
SecMax[u] = disDown[v] + ;
if(SecMax[u] > Max[u])
swap(SecMax[u],Max[u]);
}
}
disDown[u]=max(disDown[u],Max[u]);
} void dfs2(int u,int fa)
{
for(int i=,sz=G[u].size(); i<sz; i++)
{
int v = G[u][i];
if(v == fa) continue; if(disDown[v] + == Max[u])
disUp[v] = max( disUp[v] , max(disUp[u],SecMax[u])+ ); else
disUp[v] = max( disUp[v] , max(disUp[u],Max[u])+ ); dfs2(v,u);
}
} int main()
{
//freopen("E:\\acm\\input.txt","r",stdin);
cin>>n>>m>>d; memset(disDown,-0x3f,sizeof(disDown));
memset(disUp,-0x3f,sizeof(disUp));
memset(Max,-0x3f,sizeof(Max));
memset(SecMax,-0x3f,sizeof(SecMax)); for(int i=; i<=m; i++)
{
int pm;
scanf("%d",&pm);
disDown[pm] = disUp[pm] = ;
} for(int i=; i<=n; i++) G[i].clear();
for(int i=; i<n; i++)
{
int u,v;
scanf("%d %d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
} dfs1(,-);
dfs2(,-); int ans = ;
for(int i=; i<=n; i++)
if(disDown[i] <= d && disUp[i] <= d)
ans ++;
printf("%d\n",ans);
}