第一行是两个整数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。
这个是JLOI2012的T1,发出来仅为了试题完整
=============================================================================================
在这个问题中,给定一个值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。
题解:
这个题目看上去是不是要点分,稍微看一下数据范围,S不超过1000,而且所有点权都为正整数,这意味着我们每次枚举一个起点,dfs,层数不会超过1000层,而且因为要保证深度关系,很多节点都远远达不到。这题还是很暴力吧。
代码:
#include <cstdio> #include <iostream> #include <algorithm> #include <cstring> #include <cmath> #include <iostream> #define MAXN 101000 using namespace std; int ans=0; int val[MAXN],b[MAXN],dep[MAXN],roof; struct edge{ int first; int next; int to; }a[MAXN*2]; int n,m,num=0; void addedge(int from,int to){ a[++num].to=to; a[num].next=a[from].first; a[from].first=num; } void dfs(int now,int fa,int tot){ if(tot==m) ans++; if(tot>=m) return; for(int i=a[now].first;i;i=a[i].next){ int to=a[i].to; if(to==fa) continue; if(dep[to]<=dep[now]) continue; dfs(to,now,val[to]+tot); } } void pre(int now,int fa){ dep[now]=dep[fa]+1; for(int i=a[now].first;i;i=a[i].next){ int to=a[i].to;if(to==fa) continue; pre(to,now); } } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) scanf("%d",&val[i]); for(int i=1;i<=n-1;i++){ int x,y;scanf("%d%d",&x,&y); addedge(x,y),addedge(y,x);b[y]=1; } for(int i=1;i<=n;i++) if(!b[i]) roof=i; pre(roof,0); for(int i=1;i<=n;i++) dfs(i,0,val[i]); printf("%d",ans); return 0; }