HDU 3887 Counting Offspring(DFS序求子树权值和）

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<string>
#include<iostream>
#include<queue>
#include<cmath>
#include<map>
#include<stack>
#include<bitset>
using namespace std;
#define REPF( i , a , b ) for ( int i = a ; i <= b ; ++ i )
#define REP( i , n ) for ( int i = 0 ; i < n ; ++ i )
#define CLEAR( a , x ) memset ( a , x , sizeof a )
typedef long long LL;
typedef pair<int,int>pil;
const int INF = 0x3f3f3f3f;
const int maxn=1e5+100;
int L[maxn],R[maxn];
int c[maxn*2+10];
vector<int>v[maxn];
int ans[maxn];
int n,rt,dfn;
void dfs(int u,int pre)
{
    L[u]=++dfn;
    for(int i=0;i<v[u].size();i++)
    {
        int to=v[u][i];
        if(to==pre) continue;
        dfs(to,u);
    }
    R[u]=++dfn;
}
int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int val)
{
    while(x<=dfn)
    {
        c[x]+=val;
        x+=lowbit(x);
    }
}
int query(int x)
{
    int sum=0;
    while(x>0)
    {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
int main()
{
    int x,y;
    while(~scanf("%d%d",&n,&rt),n+rt)
    {
        CLEAR(L,0);
        REP(i,n+1)
          v[i].clear();
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d",&x,&y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        dfn=0;
        dfs(rt,-1);
        CLEAR(c,0);
        for(int i=1;i<=dfn;i++)
            update(i,1);
        for(int i=n;i>=1;i--)
        {
            ans[i]=(query(R[i]-1)-query(L[i]))/2;
            update(R[i],-1);
            update(L[i],-1);
        }
        REPF(i,1,n)
            printf(i==n?"%d\n":"%d ",ans[i]);
    }
    return 0;
}