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; }