#Z1656. 树上觅食

Description

小J是一只猴子，住在树上，树上每个点都有一些果子

小J每天要固定吃掉S个果子

小J还要多在树上运动一下，于是他想知道这个树上有多少路径，路径上点的果子的总和正好为S

并且路径上的点的深度是连续增加的

Format

Input

第一行给出N，S

接下来N个数字，代表每个点上的果子数

接下来N-1行，每行两个数字a,b代表a是b的父亲点

N<=100000

每个点的果子数及S均不超过int

Output

如题

Samples

输入数据 1

4 7
1 2 4 5
1 2
2 3
2 4

Copy

输出数据 1

2

---

代码

#include<bits/stdc++.h>
using namespace std;
int a[10000001],num,ans,n,x,y; 
vector<int>vec[10000001];
map<int,int>mp;
void dfs(int k,int ss)
{
	ss += a[k];
	mp[ss]++;
	if(mp.count(ss - num)) ans += mp[ss - num];
	for(int i = 0;i < vec[k].size();i++) dfs(vec[k][i],ss);
  mp[ss]--;
	if(mp[ss] == 0) mp.erase(ss);
	return ;
}
signed main()
{
	cin>>n>>num;
	for(int i = 1;i <= n;i++) cin>>a[i];
	for(int i = 1;i < n;i++)
	{
		cin>>x>>y;
		vec[x].push_back(y);
	}
	mp[0] = 1;
	dfs(1,0);
	cout<<ans<<endl;
  return 0;
}