bzoj3011[Usaco2012 Dec]Running Away From the Barn*

bzoj3011[Usaco2012 Dec]Running Away From the Barn

题意:

给出以1号点为根的一棵有边权的树,问每个点的子树中与它距离小于l的点有多少个。树的大小≤200000。

题解:

每个节点维护一个带标记可并堆,dfs时对子节点的堆加上当前节点到该子节点的边权,之后令其与当前节点的堆合并。

代码:

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #include <queue>
 5 #define inc(i,j,k) for(int i=j;i<=k;i++)
 6 #define maxn 200010
 7 #define ll long long
 8 using namespace std;
 9 
10 inline ll read(){
11     char ch=getchar(); ll f=1,x=0;
12     while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
13     while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
14     return f*x;
15 }
16 int rt[maxn],n,ch[maxn][2],ans[maxn]; ll v[maxn],tg[maxn],l;
17 struct e{int t; ll w; int n;}es[maxn]; int g[maxn],ess;
18 void pushdown(int x){
19     if(tg[x]){
20         if(ch[x][0])v[ch[x][0]]+=tg[x],tg[ch[x][0]]+=tg[x];
21         if(ch[x][1])v[ch[x][1]]+=tg[x],tg[ch[x][1]]+=tg[x];
22         tg[x]=0;
23     }
24 }
25 int merge(int x,int y){
26     if(!x||!y)return x+y; if(v[x]<v[y])swap(x,y); pushdown(x);
27     ch[x][1]=merge(ch[x][1],y); swap(ch[x][0],ch[x][1]); return x;
28 }
29 int pop(int x){
30     pushdown(x); return merge(ch[x][0],ch[x][1]);
31 }
32 void dfs(int x,int fa){
33     for(int i=g[x];i;i=es[i].n)if(es[i].t!=fa){
34         dfs(es[i].t,x); tg[rt[es[i].t]]+=es[i].w; v[rt[es[i].t]]+=es[i].w;
35         rt[x]=merge(rt[x],rt[es[i].t]); ans[x]+=ans[es[i].t];
36     }
37     while(v[rt[x]]>l)rt[x]=pop(rt[x]),ans[x]--;
38 }
39 int main(){
40     n=read(); l=read(); inc(i,2,n){int x=read(); ll y=read(); es[i-1]=(e){i,y,g[x]}; g[x]=i-1;} 
41     inc(i,1,n)rt[i]=i,v[i]=0,tg[i]=0,ans[i]=1,ch[i][0]=ch[i][1]=0; dfs(1,0);
42     inc(i,1,n)printf("%d\n",ans[i]); return 0;
43 }

 

20161025

posted @ 2016-10-30 15:48  YuanZiming  阅读(169)  评论(0编辑  收藏  举报