全WA,DFS求助!
查看原帖
全WA,DFS求助!
238361
_mxy_楼主2023/6/30 21:21
#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;
}

2023/6/30 21:21
加载中...