「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
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
#include<iostream> #include<cstdio> #include<cstring> #include<cstdlib> #include<set> #include<ctime> #include<vector> #include<cmath> #include<algorithm> #include<map> #define inf 2000000000 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; } |
Subscribe