题目:一颗树,单边修改,链上查询。。实际上链是根到结点的链。网上好像有其他做法,我的想法是这样的:
先不看修改,毫无疑问查询只是查询结点的深度;而修改一条边会有什么影响:影响是且只是以边上深度最深结点为根的子树。
所以就是DFS序了。把子树转化为区间,然后用区间修改、单点查询的线段树维护。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 255555
struct Edge{
int v,nxt;
}edge[MAXN];
int n,NE,head[MAXN];
void addEdge(int u,int v){
edge[NE].v=v; edge[NE].nxt=head[u]; head[u]=NE++;
} int dep[MAXN],l[MAXN],r[MAXN],stack[MAXN],odr;
void dfs(){
int top=;
stack[++top]=;
while(top){
int u=stack[top];
if(l[u]){
r[u]=odr; --top;
continue;
}
l[u]=++odr;
for(int i=head[u]; i!=-; i=edge[i].nxt){
int v=edge[i].v;
stack[++top]=v;
dep[v]=dep[u]+;
}
}
} int tree[MAXN<<],N,x,y;
void update(int i,int j,int k){
if(x<=i&&j<=y){
++tree[k];
return;
}
if(tree[k]){
tree[k<<]+=tree[k]; tree[k<<|]+=tree[k];
tree[k]=;
}
int mid=i+j>>;
if(x<=mid) update(i,mid,k<<);
if(y>mid) update(mid+,j,k<<|);
}
int query(int i,int j,int k){
if(i==j) return tree[k];
if(tree[k]){
tree[k<<]+=tree[k]; tree[k<<|]+=tree[k];
tree[k]=;
}
int mid=i+j>>;
if(x<=mid) return query(i,mid,k<<);
return query(mid+,j,k<<|);
} int main(){
int m,a,b;
char op[];
memset(head,-,sizeof(head));
scanf("%d",&n);
for(int i=; i<n; ++i){
scanf("%d%d",&a,&b);
if(a>b) swap(a,b);
addEdge(a,b);
}
dfs();
for(N=; N<odr; N<<=);
scanf("%d",&m);
m+=n-;
while(m--){
scanf("%s",op);
if(op[]=='W'){
scanf("%d",&a);
x=l[a];
printf("%d\n",dep[a]-query(,N,));
}else{
scanf("%d%d",&a,&b);
if(a<b) a=b;
x=l[a]; y=r[a];
update(,N,);
}
}
return ;
}