【DFS序+树状数组】HDU 3887 Counting Offspring

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 
 5 const int maxn=1e5+2;
 6 const int maxm=2*maxn; 
 7 int n,rt;
 8 struct edge
 9 {
10     int to;
11     int nxt;
12 }e[maxm];
13 int head[maxn];
14 int tot;
15 int cid;
16 int l[maxn],r[maxn]; 
17 int tree[maxm];
18 int lowbit(int x)
19 {
20     return x&(-x);
21 }
22 void add(int k,int x)
23 {
24     while(k<=cid)
25     {
26         tree[k]+=x;
27         k+=lowbit(k);
28     }
29 }
30 int query(int k)
31 {
32     int ans=0;
33     while(k)
34     {
35         ans+=tree[k];
36         k-=lowbit(k);
37     }
38     return ans;
39 }
40 void init()
41 {
42     memset(head,-1,sizeof(head));
43     tot=0;
44     cid=0;
45     memset(tree,0,sizeof(tree));
46 }
47 
48 void addedge(int u,int v)
49 {
50     e[tot].to=v;
51     e[tot].nxt=head[u];
52     head[u]=tot++;
53 }
54 void dfs(int u,int pa)
55 {
56     l[u]=++cid;
57     for(int i=head[u];i!=-1;i=e[i].nxt)
58     {
59         int v=e[i].to;
60         if(v==pa) continue;
61         dfs(v,u); 
62     }
63     r[u]=++cid;
64 } 
65 
66 
67 int main()
68 {
69     while(~scanf("%d%d",&n,&rt))
70     {
71         if(n+rt==0) break;
72         init();
73         int u,v;
74         for(int i=1;i<=n-1;i++)
75         {
76             scanf("%d%d",&u,&v);
77             addedge(u,v);
78             addedge(v,u);
79         }
80         dfs(rt,-1);
81         for(int i=1;i<=n;i++)
82         {
83             int ans=query(r[i])-query(l[i]-1);
84             if(i==1) printf("%d",ans);
85             else printf(" %d",ans);
86             add(l[i],1);
87         }
88         puts("");
89     }
90     return 0;
91 }