【图论】 树上前缀和

用set维护一下，就像这个题，因为节点深度必须是升序的，就不会出现有一条边经过根节点的点，可以用树上前缀和

问有多少条路径的节点权值总和为 ss。路径中节点的深度必须是升序的。假设节点 11 是根节点，根的深度是 00，它的儿子节点的深度为 11。路径不必一定从根节点开始。

每搜到一个点，就求一下前缀和，当搜到一个根节点再回溯的时候， 

注意一个特殊的用法set.count(const int&)

#include<iostream>
#include<cstdio>
#include<set>
using namespace std;
const int Maxn=1e5+5;
struct edge{
	int to,next;
}e[Maxn+5];
set<int> se;
int n,s,tot,ans;
int a[Maxn],head[Maxn],sum[Maxn];
inline void add(int u,int v)
{
	e[++tot].to=v;
	e[tot].next=head[u];
	head[u]=tot;
}
inline void dfs(int x,int fa)
{
	sum[x]=sum[fa]+a[x];
	se.insert(sum[x]);
	if(se.count(sum[x]-s)) ans++;
	for(int i=head[x];i;i=e[i].next)
	{
		dfs(e[i].to,x);
	}
	se.erase(sum[x]);
}
int main()
{
	scanf("%d%d",&n,&s);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1,u,v;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v);
	}
	se.insert(0);
	dfs(1,0);
	
	printf("%d",ans);
	
	
	return 0;
}