【题目链接】
【算法】
dfs序 + 线段树
树链剖分同样可以解决这个问题
【代码】
#include<bits/stdc++.h> using namespace std; #define MAXN 100010 int i,n,p,a,b,timer; int tag[MAXN<<2],sum[MAXN<<2],size[MAXN],fa[MAXN],dfn[MAXN]; vector< int > e[MAXN]; inline void dfs(int u) { int i,v; dfn[u] = ++timer; size[u] = 1; for (i = 0; i < e[u].size(); i++) { v = e[u][i]; if (fa[u] != v) { fa[v] = u; dfs(v); size[u] += size[v]; } } } inline void pushdown(int index) { if (tag[index]) { sum[index<<1] += tag[index]; sum[index<<1|1] += tag[index]; tag[index<<1] += tag[index]; tag[index<<1|1] += tag[index]; tag[index] = 0; } } inline void update(int index) { sum[index] = sum[index<<1] + sum[index<<1|1]; } inline void modify(int index,int l,int r,int ql,int qr,int val) { int mid; if (l == ql && r == qr) { sum[index] += val; tag[index] += val; return; } pushdown(index); mid = (l + r) >> 1; if (mid >= qr) modify(index<<1,l,mid,ql,qr,val); else if (mid + 1 <= ql) modify(index<<1|1,mid+1,r,ql,qr,val); else { modify(index<<1,l,mid,ql,mid,val); modify(index<<1|1,mid+1,r,mid+1,qr,val); } update(index); } inline int query(int index,int l,int r,int pos) { int mid; if (l == r) return sum[index]; pushdown(index); mid = (l + r) >> 1; if (mid >= pos) return query(index<<1,l,mid,pos); else return query(index<<1|1,mid+1,r,pos); } int main() { scanf("%d",&n); for (i = 1; i < n; i++) { scanf("%d%d",&a,&b); e[a].push_back(b); e[b].push_back(a); } dfs(1); for (i = 1; i <= n; i++) { scanf("%d",&p); printf("%d ",query(1,1,n,dfn[p])); modify(1,1,n,dfn[p],dfn[p]+size[p]-1,1); } return 0; }