【BZOJ】1782: [Usaco2010 Feb]slowdown 慢慢游

【算法】DFS序+树状数组

【题解】题意相当于统计前i-1个点在第i个点的祖先的个数，显然可以用dfs维护，用树状数组差分维护前缀和。

出栈不新加节点就要注意左闭右开，即in[a[i]]处+1,ou[a[i]]+1处-1。

出栈新加节点就要注意数组开双倍。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cctype>
#define lowbit(x) x&(-x)
using namespace std;
const int maxn=;
struct edge{int v,from;}e[maxn*];
int first[maxn],in[maxn],ou[maxn],tot=,cnt=,b[maxn*],a[maxn],n;

int read()
{
    char c;int s=,t=;
    while(!isdigit(c=getchar()))if(c=='-')t=-;
    do{s=s*+c-'';}while(isdigit(c=getchar()));
    return s*t;
}
void insert(int u,int v){tot++;e[tot].v=v;e[tot].from=first[u];first[u]=tot;}
void dfs(int x,int fa){
    in[x]=++cnt;
    for(int i=first[x];i;i=e[i].from)if(e[i].v!=fa)dfs(e[i].v,x);
    ou[x]=++cnt;
}
void modify(int x,int k){for(int i=x;i<=cnt;i+=lowbit(i))b[i]+=k;}
int ask(int x){int ans=;for(int i=x;i>=;i-=lowbit(i))ans+=b[i];return ans;}
int main(){
    n=read();
    int u,v;
    for(int i=;i<n;i++){
        u=read();v=read();
        insert(u,v);
        insert(v,u);
    }
    for(int i=;i<=n;i++)a[i]=read();
    dfs(,-);
    for(int i=;i<=n;i++){
        printf("%d\n",ask(in[a[i]]));
        modify(in[a[i]],);
        modify(ou[a[i]],-);
    }
    return ;
}