BZOJ 1782 洛谷 2982 [Usaco2010 Feb]slowdown 慢慢游

【题解】

　　一头牛走到i，相当于把i点的子树的点权都加1，查询减慢的次数就是查询目的地的点权。

　　预处理dfs序，某个点的子树的dfs序是连续的一段。差分后用树状数组维护，变成点修区查。或者直接线段树区修点查。

　　

#include<cstdio>
#include<algorithm>
#define N 200010
#define rg register
using namespace std;
int n,tot,dfn[N],size[N],p[N],t[N],last[N];
struct edge{
    int to,pre;
}e[N];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
inline void add(int x,int y){for(;x<=n;x+=x&-x) t[x]+=y;}
inline int query(int x){
    int ret=0; for(;x;x-=x&-x) ret+=t[x]; return ret;
}
void dfs(int x){
    dfn[x]=++tot; size[x]=1;
    for(rg int i=last[x],to;i;i=e[i].pre)
        if(!dfn[to=e[i].to]) dfs(to),size[x]+=size[to];
}
int main(){
    n=read();
    for(rg int i=1;i<n;i++){
        int u=read(),v=read();
        e[++tot]=(edge){v,last[u]}; last[u]=tot;
        e[++tot]=(edge){u,last[v]}; last[v]=tot;
    }
    tot=0; dfs(1);
    for(rg int i=1;i<=n;i++){
        int x=read();
        printf("%d\n",query(dfn[x]));
        add(dfn[x],1); add(dfn[x]+size[x],-1);
    }
    return 0;
}