【bzoj2783】【JLOI2012】【树】【set】
2015-08-27 20:30
351 查看
在这个问题中,给定一个值S和一棵树。在树的每个节点有一个正整数,问有多少条路径的节点总和达到S。路径中节点的深度必须是升序的。假设节点1是根节点,根的深度是0,它的儿子节点的深度为1。路径不必一定从根节点开始。
第二行是N个正整数,第i个整数表示节点i的正整数。
接下来的N-1行每行是2个整数x和y,表示y是x的儿子。
1 2 3
1 2
1 3
题解:处理一下每个点到根的路径权值和,动态的维护一个set,每dfs到一个点询问一下set里有没有当前点到根的路径权值和-s的点。有的话答案+1即可。
Input
第一行是两个整数N和S,其中N是树的节点数。第二行是N个正整数,第i个整数表示节点i的正整数。
接下来的N-1行每行是2个整数x和y,表示y是x的儿子。
Output
输出路径节点总和为S的路径数量。Sample Input
3 31 2 3
1 2
1 3
Sample Output
2HINT
对于100%数据,N≤100000,所有权值以及S都不超过1000。题解:处理一下每个点到根的路径权值和,动态的维护一个set,每dfs到一个点询问一下set里有没有当前点到根的路径权值和-s的点。有的话答案+1即可。
#include<iostream> #include<cstdio> #include<set> #define N 100010 using namespace std; set<int>s; int n,m,point ,next[N*2],k,a ,u,v,cnt,ans,deep ; struct use{ int st,en; }e[2*N]; void add(int x,int y) { next[++cnt]=point[x];point[x]=cnt; e[cnt].st=x;e[cnt].en=y; } void dfs(int x,int fa) { for (int i=point[x];i;i=next[i]) if (e[i].en!=fa) { deep[e[i].en]=deep[x]+a[e[i].en]; s.insert(deep[e[i].en]); if (s.find(deep[e[i].en]-k)!=s.end()) ans++; dfs(e[i].en,x); s.erase(deep[e[i].en]); } } int main() { scanf("%d%d",&n,&k); for (int i=1;i<=n;i++) scanf("%d",&a[i]); for (int i=1;i<=n-1;i++) { scanf("%d%d",&u,&v); add(u,v);add(v,u); } s.insert(0);deep[1]=a[1];s.insert(deep[1]);dfs(1,0); cout<<ans<<endl; }
相关文章推荐
- 函数式编程扫盲篇(转)
- SQLserver游标原理和使用方法
- jQuery学习之结构解析
- Android(java)学习笔记197:常用的对话框
- 排序算法
- Hadoop简介
- 启动另外一个Activity,并获取其返回数据
- SGU 226
- iOS多线程(GCD)(swift)
- 指针和二维数组的关系
- SQL中存储过程和自定义函数的区别
- T4M在Unity5中运行出错修改
- 3172 Virtual Friends【并查集】
- Java基于数组和循环实现的控制台五子棋
- acm hdu p2547 无剑无我{水水水}
- html中网页重定向效果
- 成员变量&&局部变量
- 好的android开发习惯
- iOS-汤姆猫项目总结
- Python语法基础——关于全局变量与局部变量