HDU - 3887 Counting Offspring（dfs序+线段树）

Counting Offspring
题 意:给你一个棵n个节点的树，f(i)定义为节点i的子树中有多少编号小于i。求f(1),f(2),…..f(n)。
数据范围:
1<=n<=1e5
输入样例:

15 7
7 10
7 1
7 9
7 3
7 4
10 14
14 2
14 13
9 11
9 6
6 5
6 8
3 15
3 12
0 0

输出数据:

0 0 0 0 0 1 6 0 3 1 0 0 0 2 0

思 路:这个挺有趣的，首先，他是对子树的操作，而且，告诉我们了，应该先从节点编号小的开始，那么我们就用线段树，先从节点编号小的开始查询，更新。
dfs序用线性的结构，维护对子树的操作。
收 获:读dfs序有了更加深刻的理解。

#include<vector>
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int maxn = 2e5+5;
vector<int> G[maxn];
int in[maxn],out[maxn];
int sum[maxn<<2];
int n,root;
int cnt;
void PushUp(int rt){
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l,int r,int rt){
    sum[rt] = 0;
    if(l == r)return;
    int m = (l+r)/2;
    build(lson);
    build(rson);
    PushUp(rt);
}
void init(){
    for(int i=1;i<=n;i++)G[i].clear();
    memset(in,0,sizeof(in));
    memset(out,0,sizeof(out));
    cnt = 0;
}
int query(int L,int R,int l,int r,int rt){
    if(L<=l && r<=R){
        return sum[rt];
    }
    int m = (l+r)/2,ans =0;
    if(L<=m) ans+=query(L,R,lson);
    if(R>m)  ans+=query(L,R,rson);
    return ans;
}
void dfs(int u,int fa){
    in[u] = ++cnt;
    for(int i=0;i<G[u].size();i++){
        int to = G[u][i];
        if(to == fa)continue;
        dfs(to,u);
    }
    out[u] = ++cnt;
}
void update(int pos,int l,int r,int rt){
    if(l == r){
        sum[rt] = 1;
        return ;
    }
    int m = (l+r)/2;
    if(pos<=m) update(pos,lson);
    else update(pos,rson);
    PushUp(rt);
}
int main(){
    while(~scanf("%d %d",&n,&root) && (n+root)!=0){
        init();
        int u,v;
        for(int i=0;i<n-1;i++){
            scanf("%d %d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs(root,-1);
        build(1,2*n,1);
        int ans;
        for(int i=1;i<=n;i++){
            int l = in[i],r = out[i];
            ans = query(l,r,1,2*n,1);
            update(l,1,2*n,1);
            printf("%d%c",ans,i==n?'\n':' ');
        }
    }
    return 0;
}