社区讨论

全WA,DFS求助!

P5908猫猫和企鹅参与者 2已保存回复 4

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
4 条
当前快照
1 份
快照标识符
@lo2d5034
此快照首次捕获于
2023/10/23 11:53
2 年前
此快照最后确认于
2023/11/03 12:02
2 年前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
#define long long ll
const int MAXN=1e5+10;

int n,d;
struct edge
{
	int u,v;
	int nxt;
} edges[100010];
int cnt=0,head[100010];
int sum=0;
bool vis[100010];

void add(int u,int v)
{
	edges[++cnt].u=u;
	edges[cnt].v=v;
	edges[cnt].nxt=head[u];
	head[u]=cnt;
}
void dfs(int u,int step)
{
	if(step>d)
	{
		return;
	}
	for(int i=head[u]; i; i=edges[i].nxt)
	{
		if(!vis[edges[i].v])
		{
			sum++;
			vis[edges[i].v]=1;
			dfs(edges[i].v,step+1);
		}
	}
	return;
}

int main()
{
	cin>>n>>d;
	for(int i=1; i<n; i++)
	{
		int u,v;
		cin>>u>>v;
		add(u,v);
	}

	dfs(1,1);

	cout<<sum<<endl;
	return 0;
}

回复

4 条回复,欢迎继续交流。

正在加载回复...