题目描述
传送门
题目大意:给出一颗n个节点的树,按照一定的顺序激活每个点,并询问每个点到根的路径上经过了多少个已经激活的点。
题解
这题应该是一眼树链剖分吧。不过拿树链剖分来做其实有点大材小用了。直接线段树+DFS就可以。
每次单点查询,将子树对应的dfs序中的区间+1.
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 200003
using namespace std;
int n,point[N],nxt[N],v[N],l[N],pos[N],r[N];
int tr[N*4],delta[N*4],tot,sz;
void add(int x,int y)
{
tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
tot++; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int fa)
{
l[x]=++sz; pos[sz]=x;
for (int i=point[x];i;i=nxt[i]) {
if (v[i]==fa) continue;
dfs(v[i],x);
}
r[x]=sz;
}
void pushdown(int now)
{
if (delta[now]) {
tr[now<<1]+=delta[now]; tr[now<<1|1]+=delta[now];
delta[now<<1]+=delta[now]; delta[now<<1|1]+=delta[now];
delta[now]=0;
}
}
int find(int now,int l,int r,int x)
{
if (l==r) return tr[now];
int mid=(l+r)/2;
pushdown(now);
if (x<=mid) return find(now<<1,l,mid,x);
else return find(now<<1|1,mid+1,r,x);
}
void qjchange(int now,int l,int r,int ll,int rr)
{
if (ll<=l&&r<=rr) {
tr[now]++; delta[now]++;
return;
}
int mid=(l+r)/2;
pushdown(now);
if (ll<=mid) qjchange(now<<1,l,mid,ll,rr);
if (rr>mid) qjchange(now<<1|1,mid+1,r,ll,rr);
}
int main()
{
freopen("a.in","r",stdin);
scanf("%d",&n);
for (int i=1;i<n;i++) {
int x,y; scanf("%d%d",&x,&y);
add(x,y);
}
dfs(1,0);
for (int i=1;i<=n;i++) {
int x; scanf("%d",&x);
printf("%d\n",find(1,1,n,l[x]));
qjchange(1,1,n,l[x],r[x]);
}
}