help 满江红bfs代码
查看原帖
help 满江红bfs代码
1043012
Furukawa_Nagisa楼主2024/1/20 20:29

全错 求助awa

#include<bits/stdc++.h>
using namespace std;
int n,fa[250000],root,sum=0,s,d;
vector<int>son[250000];queue<int>v;
void bfs(int root){
	v.push(root);s=1;
	while(s<d+1){
		int faa=v.front();v.pop();s++;
		for(int i=0;i<son[faa].size();i++){
			v.push(son[faa][i]);sum++;
		}
	}
	cout<<sum;
}
int main(){
	cin>>n>>d;
	for(int i=1;i<n;i++){
		int si,fi;cin>>fi>>si;
		if(si==1){
			si=fi;fi=si;
		}
		son[fi].push_back(si);
		fa[si]=fi;
	}
	bfs(1);
	return 0;
}
2024/1/20 20:29
加载中...