社区讨论
已AC求助
CF337DBook of Evil参与者 1已保存回复 0
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 0 条
- 当前快照
- 1 份
- 快照标识符
- @mi5yzh0o
- 此快照首次捕获于
- 2025/11/19 20:19 4 个月前
- 此快照最后确认于
- 2025/11/21 00:01 4 个月前
3遍dfs,想知道倒数第4行为啥dep数组要-1才是对的
CPP#include<iostream>
#include<stdio.h>
#define N 100005
using namespace std;
int n,m,d,a[N];
int cnt,head[N],to[N<<2],nxt[N<<2];
void add(int u,int v)
{
to[++cnt]=v;
nxt[cnt]=head[u];
head[u]=cnt;
}
int dep1[N],dep2[N],c,ans;
void dfs1(int x,int fa)
{
dep1[x]=dep1[fa]+1;
for(int i=head[x];i;i=nxt[i])
{
int y=to[i];
if(y==fa)
continue;
dfs1(y,x);
}
}
void dfs2(int x,int fa)
{
dep2[x]=dep2[fa]+1;
for(int i=head[x];i;i=nxt[i])
{
int y=to[i];
if(y==fa)
continue;
dfs2(y,x);
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m>>d;
for(int i=1;i<=m;i++)
cin>>a[i];
for(int i=1;i<n;i++)
{
int u,v;
cin>>u>>v;
add(u,v);
add(v,u);
}
dfs1(1,0);
for(int i=1;i<=m;i++)
if(dep1[a[i]]>dep1[c])
c=a[i];
dfs2(c,0);
c=0;
for(int i=1;i<=m;i++)
if(dep2[a[i]]>dep2[c])
c=a[i];
dfs1(c,0);
for(int i=1;i<=n;i++)
if(dep1[i]-1<=d&&dep2[i]-1<=d)
ans++;
cout<<ans<<"\n";
return 0;
}
回复
共 0 条回复,欢迎继续交流。
正在加载回复...