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

http://acm.hdu.edu.cn/showproblem.php?pid=3887

【题意】

• 给定一棵树，给定这棵树的根
• 对于每个结点，统计子树中编号比他小的结点个数
• 编号从小到大一次输出

【思路】

• 从小到大处理每个结点，即统计当前结点的结果后，把当前结点插入到树状数组中

【AC】

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;

 const int maxn=1e5+;
 const int maxm=*maxn;
 int n,rt;
 struct edge
 {
     int to;
     int nxt;
 }e[maxm];
 int head[maxn];
 int tot;
 int cid;
 int l[maxn],r[maxn];
 int tree[maxm];
 int lowbit(int x)
 {
     return x&(-x);
 }
 void add(int k,int x)
 {
     while(k<=cid)
     {
         tree[k]+=x;
         k+=lowbit(k);
     }
 }
 int query(int k)
 {
     int ans=;
     while(k)
     {
         ans+=tree[k];
         k-=lowbit(k);
     }
     return ans;
 }
 void init()
 {
     memset(head,-,sizeof(head));
     tot=;
     cid=;
     memset(tree,,sizeof(tree));
 }

 void addedge(int u,int v)
 {
     e[tot].to=v;
     e[tot].nxt=head[u];
     head[u]=tot++;
 }
 void dfs(int u,int pa)
 {
     l[u]=++cid;
     for(int i=head[u];i!=-;i=e[i].nxt)
     {
         int v=e[i].to;
         if(v==pa) continue;
         dfs(v,u);
     }
     r[u]=++cid;
 } 

 int main()
 {
     while(~scanf("%d%d",&n,&rt))
     {
         if(n+rt==) break;
         init();
         int u,v;
         for(int i=;i<=n-;i++)
         {
             scanf("%d%d",&u,&v);
             addedge(u,v);
             addedge(v,u);
         }
         dfs(rt,-);
         for(int i=;i<=n;i++)
         {
             int ans=query(r[i])-query(l[i]-);
             if(i==) printf("%d",ans);
             else printf(" %d",ans);
             add(l[i],);
         }
         puts("");
     }
     return ;
 } 

树状数组