bzoj2783 [JLOI2012]树 STL+dfs

题意：给出一棵树。每个点有权值，问有多少条路径的权值和=S，S给定
S<=1e3,N<=1e5.
第一反应dp，设f[i][j]表示第i个点，权值和为j，dp式十分显然= =然而会T，卡了卡常卡不过去。。应该是卡掉了这种做法。。
那么我们只能nlogn，注意到答案好像不用mo，然后我觉得可能时间复杂度是O（ans）。。然后dfs一下用个set存储一下前缀和，然后每次查找一下是否有一条合法路径。。也可不用set，直接二分查找。
查找sum[x]-k这个也很容易理解啦，只要存在sum[x]-k,又存在sum[x]，那么肯定有一段路径和为k，防止计重遍历完这一颗子树就把这个点删掉。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<set>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
int n,m,a[N],head[N<<1],next[N<<1],go[N<<1];
int sum[N],tot,ans;
multiset<int> q;
inline void add(int x,int y)
{
    go[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}
inline void dfs(int x)
{
    if (q.find(sum[x]-m)!=q.end())ans++;
    q.insert(sum[x]);
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        sum[v]=sum[x]+a[v];
        dfs(v);
    }
    q.erase(q.find(sum[x]));
}
int main()
{
    scanf("%d%d",&n,&m);
    q.insert(0);
    fo(i,1,n)scanf("%d",&a[i]);
    fo(i,1,n-1)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    sum[1]=a[1];
    dfs(1);
    printf("%d\n",ans);
}