bzoj 1782: [Usaco2010 Feb]slowdown 慢慢游 （线段树+DFS序）

题目描述

传送门

题目大意：给出一颗n个节点的树，按照一定的顺序激活每个点，并询问每个点到根的路径上经过了多少个已经激活的点。

题解

这题应该是一眼树链剖分吧。不过拿树链剖分来做其实有点大材小用了。直接线段树+DFS就可以。
每次单点查询，将子树对应的dfs序中的区间+1.

代码

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>#define N 200003using 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]);    }}