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
,head[N<<1],next[N<<1],go[N<<1];
int sum
,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);
}