C++(clang++ 11.0.1) 解法, 执行用时: 37ms, 内存消耗: 1204K, 提交时间: 2022-11-18 18:53:17

#include<bits/stdc++.h>
using namespace std;
struct tree{
	int val,fa;
}a[100010];
int n,root,ans=0,s,len;
void work(){
	int k; 
	for(int i=1;i<=n;i++){
		len=0;
		int x=0;
		k=i;
		while(k!=root){
			len++;
			x+=a[k].val;
			k=a[k].fa;
			if(x==s){ans++;break;}
			if(x>s){
				break;//TODO
			}
			//TODO
		}
		x+=a[k].val;
		if(x==s)ans++;//TODO
	}
}

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin>>n>>s;
	for(int i=1;i<=n;i++){
		cin>>a[i].val;//TODO
	}
	for(int i=1;i<=n-1;i++){
		int x,y;
		cin>>x>>y;
		a[y].fa=x;//TODO
	}
	for(int i=1;i<=n;i++){
		if(a[i].fa==0){
		root=i;
		break;	
		}
		//TODO
	}
	work();
	cout<<ans;
	return 0;
}