全RE求调
查看原帖
全RE求调
763878
Jerry_heng楼主2024/3/11 19:13
#include<bits/stdc++.h>
using namespace std;
const int mxn=2500010;
int head[mxn],dfn[mxn],dep[mxn],cnt,siz[mxn],n,m,cnt2,tree[mxn<<2];
struct node{
	int to,nxt;
}edge[mxn<<1];
void add(int u,int v){
	edge[++cnt]=(node){v,head[u]};
	head[u]=cnt;
}
void dfs(int u,int f){
	siz[u]=1;
	dep[u]=dep[f]+1;
	dfn[u]=++cnt2;
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		dfs(v,u);
		siz[u]+=siz[v];
	}
}
void pushdown(int o){
	if(tree[o]){
		tree[o*2]+=tree[o];
		tree[o*2+1]+=tree[o];
		tree[o]=0;
	}
}
void update(int o,int l,int r,int L,int R){
	if(L<=l&&r<=R){
		tree[o]++;
		return;
	}
	pushdown(o);
	int mid=(l+r)>>1;
	if(L<=mid)update(o*2,l,mid,L,R);
	if(R>mid)update(o*2+1,mid+1,r,L,R);
}
int query(int o,int l,int r,int pos){
	if(l==r)return tree[o];
	int mid=(l+r)>>1;
	pushdown(o);
	if(pos<=mid)query(o*2,l,mid,pos);
	else query(o*2+1,mid+1,r,pos);
}
int main(){
	ios::sync_with_stdio(0);
	cin>>n;
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		if(u>v)swap(u,v);
		add(u,v);
	}
	dep[0]=-1;
	dfs(1,0);
	cin>>m;
	m=m+n-1;
	while(m--){
		string op;
		int u,v;
		cin>>op;
		if(op=="W"){
			cin>>u;
			cout<<dep[u]-query(1,1,n,dfn[u])<<endl;
		}
		else{
			cin>>u>>v;
			if(u<v)swap(u,v);
			update(1,1,n,dfn[u],dfn[u]+siz[u]-1);
		}
	}
	return 0;
}
2024/3/11 19:13
加载中...