hdu3887 Counting Offspring(dfs序+树状数组)

题目链接：点击打开链接

题目描述：给定一棵树，统计一棵子树中节点编号小于子树根节点的数量？

解题思路：

先dfs预处理，对于每个子树可以得到一段连续的序列

使用树状数组求连续区间的和

注意：

1、树是无向图，建树时注意建两条边，由于测试案例比较坑，我以为左边为右边的父节点，所以建图时只见了一条边，导致一直tle

2、dfs容易栈溢出，记得使用stack模拟递归（stack+标记数组即可）

3、注意输出格式

#include <cstdio>
#include <cstring>
#include <stack>
#define MAXN 100010
using namespace std;
int vis[MAXN][2];
int n,p;
int head[MAXN];
struct Edge{
    int v,next;
}edge[MAXN*2];///Time Limit Exceeded
int tol;
void addEdge(int u,int v){
    edge[tol].v=v; edge[tol].next=head[u]; head[u]=tol++;
    edge[tol].v=u; edge[tol].next=head[v]; head[v]=tol++;///Time Limit Exceeded
}
int id;
stack<int> st;
bool flag[MAXN];
void dfs(int rt){
    while(!st.empty()) st.pop();
    vis[rt][0]=id++;
    st.push(rt);
    flag[rt]=true;
    while(!st.empty()){
        rt=st.top();
        while(flag[edge[head[rt]].v]) head[rt]=edge[head[rt]].next;///Time Limit Exceeded
        if(head[rt]!=-1){
            vis[edge[head[rt]].v][0]=id++;
            flag[edge[head[rt]].v]=true;
            st.push(edge[head[rt]].v);
            head[rt]=edge[head[rt]].next;
        }
        else
        {
            vis[rt][1]=id++;
            st.pop();
        }
    }
}
int C[MAXN*2];
int lowbit(int x){
    return x&(-x);
}
int sum(int pos){
    int ret=0;
    while(pos>0){
        ret+=C[pos];
        pos-=lowbit(pos);
    }
    return ret;
}
void add(int pos,int v){
    while(pos<=id){
        C[pos]+=v;
        pos+=lowbit(pos);
    }
}
int main(){
    while(scanf("%d%d",&n,&p)!=EOF&&(n!=0||p!=0)){
        tol=0; memset(head,-1,sizeof(head));
        int u,v;
        for(int i=0;i<n-1;++i){scanf("%d%d",&u,&v); addEdge(u,v); }
        memset(flag,false,sizeof(flag));
        id=1;dfs(p);
        id--;
        memset(C,0,sizeof(C));
        for(int i=1;i<n;i++){
            add(vis[i][0],1);
            add(vis[i][1],1);
            printf("%d ",(sum(vis[i][1]-1)-sum(vis[i][0]))/2);
        }
        add(vis[n][0],1);
        add(vis[n][1],1);
        printf("%d\n",(sum(vis[n][1]-1)-sum(vis[n][0]))/2);///Presentation Error
    }
    return 0;
}