題意:樹上有一個點可以覆蓋距離離他d的所有點,告訴你已有多少點被覆蓋,問這個點可能有幾種情況
思路:從根開始搜索 構造一個數組記錄深度比他深的 且已被覆蓋 距離他最遠的節點的距離 然後再進入搜索對於每個節點 拿到父節點離他最遠的被覆蓋節點的距離(不在這個節點下面) 即可求出答案
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<stdio.h>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define N 100010
#define MOD 1000000009
#define LL long long
vector<int> e[N];
int dep[N];
int v[N];
int g[N];
int ans,n,m,k;
void dfs(int x,int fa)
{
for(int i=0;i<e[x].size();i++)
{
int to=e[x][i];
if(to==fa)continue;
dfs(to,x);
if(dep[to]>0||v[to])
dep[x]=max(dep[x],dep[to]+1);
}
}
void solve(int x,int fa,int dis)
{
if(k>=max(dis,dep[x]))ans++;
int l=0,r=0,a=0,b=0;
for(int i=0;i<e[x].size();i++)
{
int to=e[x][i];
if(to==fa)continue;
g[to]=max(dis,l>0?l+1:a);
l=max(l,dep[to]);
a|=v[to];
}
for(int i=e[x].size()-1;i>=0;i--)
{
int to=e[x][i];
if(to==fa)continue;
g[to]=max(g[to],r>0?r+1:b);
r=max(r,dep[to]);
b|=v[to];
}
for(int i=0;i<e[x].size();i++)
{
int to=e[x][i];
if(to==fa)continue;
if(g[to]>0||v[x])
solve(to,x,g[to]+1);
else
solve(to,x,0);
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
int x,y;
for(int i=0;i<m;i++)
{
scanf("%d",&x);
v[x]=1;
}
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
e[x].push_back(y);
e[y].push_back(x);
}
dfs(1,1);
solve(1,1,0);
printf("%d\n",ans);
return 0;
}