全WA,自己构造的数据都能过,求助啊
查看原帖
全WA,自己构造的数据都能过,求助啊
79543
一大楼主2023/7/21 16:07
#include<bits/stdc++.h>
using namespace std;

const int m=200500;
int n,d;
struct edge{
	int next,to;
}a[m];
int head[m],cnt;
void link(int u,int v){
	a[++cnt].next=head[u];
	head[u]=cnt;
	a[cnt].to=v;	
}
int v[m],dis[m];
int dfs(int x){
	int tot=1;
	if(v[x]==d) return tot;
	for(int i=head[x];i;i=a[i].next){
		int y=a[i].to;
		if(!v[y]&&y!=1){
			v[y]=v[x]+1;
			tot+=dfs(y);
		}
	}
	return tot;
}
int main(){
	cin>>n>>d;
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		link(u,v);
		link(v,u);
	}
	v[1]=0;
	
	cout<<dfs(1)-1;
} 
2023/7/21 16:07
加载中...