社区讨论

已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 条回复,欢迎继续交流。

正在加载回复...