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=,f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',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=; for(;x;x-=x&-x) ret+=t[x]; return ret;
 }
 void dfs(int x){
     dfn[x]=++tot; size[x]=;
     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=;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=; dfs();
     for(rg int i=;i<=n;i++){
         int x=read();
         printf("%d\n",query(dfn[x]));
         add(dfn[x],); add(dfn[x]+size[x],-);
     }
     return ;
 }