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;
}