0pts,求调,玄关
查看原帖
0pts,求调,玄关
1125129
WJX120423楼主2024/8/17 13:43
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int ans=0;
int n,s,x,y;
struct node{
	int val;
	vector<int> v;
}e[maxn];
struct qnode{
	int id,sum;
	qnode(int d,int su){
		id=d; sum=su;
	}
};
queue<qnode> q;
void bfs(int ss){
	q.push(qnode(s,e[ss].val));
	while(!q.empty()){
		qnode nd=q.front(); q.pop();
		if(nd.sum==s){
			//cout<<nd.sum<<endl;
			ans++;
			continue;
		}
		for(int i=0;i<e[nd.id].v.size();i++){
			q.push(qnode(e[nd.id].v[i],nd.sum+e[e[nd.id].v[i]].val));
		}
	}
}
int main(){
	cin>>n>>s;
	for(int i=1;i<=n;i++) cin>>e[i].val;
	for(int i=1;i<=n-1;i++){
		cin>>x>>y;
		e[x].v.push_back(y);
	}
	for(int i=1;i<=3;i++) bfs(i);
	cout<<ans;
	return 0;
}
2024/8/17 13:43
加载中...