bzoj 2783 树(dfs 前缀和 邻接表)(中等)

【bzoj2783】[JLOI2012]树

Description

第一行是两个整数N和S，其中N是树的节点数。

第二行是N个正整数，第i个整数表示节点i的正整数。

接下来的N-1行每行是2个整数x和y，表示y是x的儿子。

输出格式：

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

 

输入样例：输出样例：3 3

1 2 3

1 2

1 3

2

 

数据范围：

对于30%数据，N≤100；

对于60%数据，N≤1000；

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

=============================================================================================

在这个问题中，给定一个值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。记录元素前缀和。

比如搜到x，在set里面查询有没有s[x]-k，有的话ans++，然后继续往下dfs时插入s[x]

最开始要插入0

#include<iostream>#include<cstdio>#include<set>using namespace std;inline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int n,s,cnt,ans;int v[100005],last[100005],sum[100005];multiset<int> q;struct edge{int to,next;}e[100005];void insert(int u,int v){e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;}void dfs(int x){if(q.find(sum[x]-s)!=q.end())        ans++;q.insert(sum[x]);for(int i=last[x];i;i=e[i].next){sum[e[i].to]=sum[x]+v[e[i].to];dfs(e[i].to);}q.erase(q.find(sum[x]));}int main(){n=read();s=read();q.insert(0);for(int i=1;i<=n;i++)v[i]=read();for(int i=1;i<n;i++){int u=read(),v=read();insert(u,v);}sum[1]=v[1];dfs(1);printf("%d\n",ans);return 0;}