树

jzoj 2753

题目大意：

给你一棵树，每一个点都一个值，现在问你有多少条路径可以满足以下条件：
1、方向都是向下
2、路径上的点的值总和为S

输入样例

3 3
1 2 3
1 2
1 3

输出样例

2

数据范围

对于30%数据，$N⩽100$；
对于60%数据，$N⩽1000$；
对于100%数据，$N⩽100000$，所有权值以及S都不超过1000。

解题思路：

从根节点开始搜索，用一个值num来记录根节点到当前点的值总和
如果以某一个点为路径开始，拿如果下面有某一个点的num为当前点的num加S那这就是一条合法路径

代码：

#include<map> 
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
int n, s, x, y, ans, tot, b[105000], head[105000];
map< int , int >p;
struct rec
{int to, next;
}a[105000];
void add(int x, int y)
{a[++tot].to = y;a[tot].next = head[x];head[x] = tot;
}
void dfs(int x, int num)
{int k = p[num + s];//记录num为这个值的点有多少个p[num + b[x]]++;//多了一个for (int i = head[x]; i; i = a[i].next)dfs(a[i].to, num + b[x]);//搜索ans += p[num + s] - k;//看多了多少
}
int main()
{scanf("%d%d", &n, &s);for (int i = 1; i <= n; ++i)scanf("%d", &b[i]);for (int i = 1; i < n; ++i){scanf("%d%d", &x, &y);add(x, y);//连边}dfs(1, 0);printf("%d", ans);return 0;
}