【JLOI2012】【BZOJ2783】树

Description
在这个问题中，给定一个值S和一棵树。在树的每个节点有一个正整数，问有多少条路径的节点总和达到S。路径中节点的深度必须是升序的。假设节点1是根节点，根的深度是0，它的儿子节点的深度为1。路径不必一定从根节点开始。
Input

   第一行是两个整数N和S，其中N是树的节点数。
   第二行是N个正整数，第i个整数表示节点i的正整数。
   接下来的N-1行每行是2个整数x和y，表示y是x的儿子。

Output

   输出路径节点总和为S的路径数量。

Sample Input

3 3

1 2 3

1 2

1 3

Sample Output

2
HINT

对于100%数据，N≤100000，所有权值以及S都不超过1000。

由于路径必须是深到浅的一条链
所以DFS+set维护出现过的值就行了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#define MAXN 100010
using namespace std;
int n,s,top;
int val[MAXN],dis[MAXN],fa[MAXN];
int root;
long long ans;
multiset<int> S;
struct edge
{
    int to;
    edge *next;
}e[MAXN<<1],*prev[MAXN];
void insert(int u,int v)
{
    e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top];
}
void dfs(int x)
{
    for (edge *i=prev[x];i;i=i->next)
    {
        dis[i->to]=dis[x]+val[i->to];
        S.insert(dis[i->to]);
        if (S.find(dis[i->to]-s)!=S.end())  ans++;
        dfs(i->to);
        S.erase(dis[i->to]);
    }
}
int main()
{
    scanf("%d%d",&n,&s);S.clear();
    for (int i=1;i<=n;i++)  scanf("%d",&val[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        fa[y]=x;insert(x,y);
    }
    for (int i=1;i<=n;i++)
        if (!fa[i]) root=i;
    dis[root]=val[root];S.insert(0);S.insert(val[root]);
    if (S.find(dis[root]-s)!=S.end())   ans++;
    dfs(root);
    printf("%lld\n",ans);
}