luogu 2982 [USACO10FEB]慢下来Slowing down dfs序+树状数组

将要查询的信息放到 dfs 序上并用树状数组查一个前缀和即可. 

#include <bits/stdc++.h>  
#define N 100004 
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;  
int n,edges,tim; 
int hd[N],to[N<<1],nex[N<<1],st[N],ed[N],C[N]; 
void add(int u,int v) 
{
    nex[++edges]=hd[u],hd[u]=edges,to[edges]=v; 
}  
int lowbit(int t) 
{
    return t&(-t); 
} 
void update(int x,int d) 
{ 
    for(;x<N;x+=lowbit(x)) C[x]+=d; 
} 
int query(int x) 
{
    int tmp=0; 
    for(;x>0;x-=lowbit(x)) tmp+=C[x]; 
    return tmp; 
}
void dfs(int u,int ff) 
{ 
    st[u]=++tim;    
    for(int i=hd[u];i;i=nex[i]) 
    {
        int v=to[i]; 
        if(v!=ff) dfs(v,u); 
    } 
    ed[u]=tim;  
}
int main() 
{  
    int i,j; 
    //setIO("input");  
    scanf("%d",&n);  
    for(i=1;i<n;++i) 
    {
        int u,v; 
        scanf("%d%d",&u,&v),add(u,v),add(v,u); 
    } 
    dfs(1,0);        
    for(i=1;i<=n;++i) 
    {  
        int p; 
        scanf("%d",&p);       
        printf("%d\n",query(st[p]));      
        update(st[p],1); 
        update(ed[p]+1,-1);        
    }
    // printf("%lld\n",ans); 
    return 0; 
}