bzoj2783 [JLOI2012]树
Posted wfj_2048
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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。
这个是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
1 2 3
1 2
1 3
Sample Output
HINT
对于100%数据,N≤100000,所有权值以及S都不超过1000。
正解:$set$启发式合并。
题面鬼得不行。。
比较简单的一个题,直接在树上用启发式合并即可,我偷懒所以写了个$multiset$。。
1 #include <bits/stdc++.h> 2 #define il inline 3 #define RG register 4 #define ll long long 5 #define N (200010) 6 7 using namespace std; 8 9 struct edge{ int nt,to; }g[N<<1]; 10 11 int head[N],val[N],son[N],sz[N],rt[N],tim[N],num,n,S; 12 ll ans; 13 14 multiset<int> st[N]; 15 16 multiset<int>:: iterator it; 17 18 il int gi(){ 19 RG int x=0,q=1; RG char ch=getchar(); 20 while ((ch<‘0‘ || ch>‘9‘) && ch!=‘-‘) ch=getchar(); 21 if (ch==‘-‘) q=-1,ch=getchar(); 22 while (ch>=‘0‘ && ch<=‘9‘) x=x*10+ch-48,ch=getchar(); 23 return q*x; 24 } 25 26 il void insert(RG int from,RG int to){ 27 g[++num]=(edge){head[from],to},head[from]=num; return; 28 } 29 30 il void dfs(RG int x,RG int p){ 31 sz[x]=1; RG int v,tr; 32 for (RG int i=head[x];i;i=g[i].nt){ 33 v=g[i].to; if (v==p) continue; 34 dfs(v,x),sz[x]+=sz[v]; 35 if (sz[son[x]]<=sz[v]) son[x]=v; 36 } 37 if (val[x]==S) ++ans; 38 if (!son[x]){ rt[x]=x,st[x].insert(val[x]); return; } 39 rt[x]=rt[son[x]],tim[rt[x]]+=val[x],tr=rt[x]; 40 ans+=st[tr].count(S-tim[tr]); 41 for (RG int i=head[x],Rt,d;i;i=g[i].nt){ 42 v=g[i].to; if (v==p || v==son[x]) continue; Rt=rt[v]; 43 for (it=st[Rt].begin();it!=st[Rt].end();++it){ 44 d=(*it)+tim[Rt]+val[x]; if (d==S) ++ans; 45 if (d<=S) st[tr].insert(d-tim[tr]); 46 } 47 st[Rt].clear(); 48 } 49 st[tr].insert(val[x]-tim[tr]); return; 50 } 51 52 int main(){ 53 #ifndef ONLINE_JUDGE 54 freopen("tree.in","r",stdin); 55 freopen("tree.out","w",stdout); 56 #endif 57 n=gi(),S=gi(); 58 for (RG int i=1;i<=n;++i) val[i]=gi(); 59 for (RG int i=1,x,y;i<n;++i) 60 x=gi(),y=gi(),insert(x,y),insert(y,x); 61 dfs(1,0),cout<<ans; return 0; 62 }
以上是关于bzoj2783 [JLOI2012]树的主要内容,如果未能解决你的问题,请参考以下文章