WA树链剖分90pts求调
查看原帖
WA树链剖分90pts求调
605945
luo_xiaoran楼主2024/3/20 12:24
#include<bits/stdc++.h>
#define int long long
#define ll long long
#define N (int)3e5+5
using namespace std;
int n,m,root=1,cnt,tot,it,b[N][2],a[N],head[N],dep[N],sz[N],son[N],id[N],top[N],fat[N];
ll na[N],tr[N<<2],la[N<<2];
struct Edge{
	int u,v,nxt;
}e[N<<1];
inline void add(int u,int v){
	e[++cnt]=Edge{u,v,head[u]};
	head[u]=cnt;
}
inline void dfs1(int u,int fa){
	dep[u]=dep[fa]+1;sz[u]=1;fat[u]=fa;
	for(int i=head[u];i;i=e[i].nxt){
		if(e[i].v==fa) continue;
		fat[e[i].v]=u;
		dfs1(e[i].v,u);
		sz[u]+=sz[e[i].v];
		if(sz[e[i].v]>sz[son[u]]) son[u]=e[i].v;
	}
}
inline void dfs2(int u,int fa){
	id[u]=++tot;
	if(son[u]) top[son[u]]=top[u],dfs2(son[u],u);
	for(int i=head[u];i;i=e[i].nxt){
		if(e[i].v==fa||e[i].v==son[u]) continue;
		top[e[i].v]=e[i].v;
		dfs2(e[i].v,u);
	}
}
inline void pushup(int p){
	tr[p]=tr[p<<1]+tr[p<<1|1];
}
//inline void build(int l,int r,int p){
//	if(l==r){
//		tr[p]=na[l];
//		return ;
//	}
//	int m=(l+r)>>1;
//	build(l,m,p<<1);build(m+1,r,p<<1|1);
//	pushup(p);
//}
inline void pushdown(int s,int t,int p){
	int m=s+t>>1;
	if(la[p]) tr[p<<1]+=la[p],tr[p<<1|1]+=la[p],la[p<<1]+=la[p],la[p<<1|1]+=la[p];
	la[p]=0;
}
inline void update(int l,int r,int s,int t,int c,int p){
	if(l>r) return ;
	pushdown(s,t,p);
	if(l<=s&&t<=r){
		tr[p]+=c;
		la[p]+=c;
		return ;
	}
	pushdown(s,t,p);
	int m=(s+t)>>1;
	if(l<=m) update(l,r,s,m,c,p<<1);
	if(r>m) update(l,r,m+1,t,c,p<<1|1);
	pushup(p);
}
inline ll query(int l,int r,int s,int t,int p){
	if(l>r) return 0;
	if(l<=s&&t<=r) return tr[p];
	pushdown(s,t,p);
	ll m=(s+t)>>1,v=0;
	if(l<=m) v=query(l,r,s,m,p<<1);
	if(r>m) v+=(query(l,r,m+1,t,p<<1|1));
	return v;
}
void update_path(int u,int v,int c){
	while(top[u]^top[v]){
		if(dep[top[u]]<dep[top[v]]) u^=v^=u^=v;
		update(id[top[u]],id[u],1,n,c,1);
		u=fat[top[u]];
	}
	if(dep[u]<dep[v]) u^=v^=u^=v;
	update(id[v]+1,id[u],1,n,c,1);
}
ll query_path(int u,int v){
	ll ans=0;
	while(top[u]^top[v]){
		if(dep[top[u]]<dep[top[v]]) u^=v^=u^=v;
		ans+=query(id[top[u]],id[u],1,n,1);
		u=fat[top[u]];
	}
	if(dep[u]<dep[v]) u^=v^=u^=v;
	return ans+query(id[v]+1,id[u],1,n,1);
}
signed main(){
	cin>>n>>m;
	for(int i=1,u,v;i<n;++i) scanf("%lld%lld",&u,&v),add(u,v),add(v,u);
	top[root]=root;
	dfs1(root,0);
	dfs2(root,0);
	while(m--){
		char s[8];
		int u,v;
		scanf("%s",s);
		scanf("%lld",&u);
		if(s[0]=='Q'){
			scanf("%lld",&v);
			if(query_path(u,v)) printf("No\n");
			else printf("Yes\n");
		}
		else if(s[0]=='C'){
			scanf(" %lld",&v);
			b[++it][0]=u,b[it][1]=v;
			update_path(u,v,1);
		}
		else if(s[0]=='U') update_path(b[u][0],b[u][1],-1);
	}
	return 0;
}

WA#10

2024/3/20 12:24
加载中...