qwq
  • 板块学术版
  • 楼主wuzhirui
  • 当前回复6
  • 已保存回复6
  • 发布时间2024/1/22 14:29
  • 上次更新2024/1/22 17:06:56
查看原帖
qwq
1046964
wuzhirui楼主2024/1/22 14:29
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn=100000+100;
struct node{
	int to,next,id,val;
}e[maxn*2];
struct tree{
	int l,r,sum,add;
}tr[maxn*4];
int head[maxn],k,cnt;
void add(int u,int v,int val){
	e[++k].to=v;
	e[k].next=head[u];
	e[k].val=val;
	head[u]=k;
}
int dep[maxn],f[maxn],son[maxn],siz[maxn],w[maxn];
void dfs1(int u,int fa,int d){
	f[u]=fa;dep[u]=d;siz[u]=1;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(fa==v) continue;
		w[v]=e[i].val;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];
		if(son[u]==0 || siz[son[u]]<siz[v]) son[u]=v;
	}
}
int b[maxn],top[maxn],pos[maxn],a[maxn];
void dfs2(int u,int tp){
	top[u]=tp;pos[u]=++cnt;a[cnt]=w[u];
	if(son[u]!=0) return ;
	dfs2(son[u],tp);
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(v!=f[u] && v!=son[u]) dfs2(v,v);
	}
}
void build(int id,int l,int r){
	tr[id].l=l;tr[id].r=r;tr[id].sum=0;
	if(l==r){
		tr[id].sum=a[l];
		return ;
	}
	int mid=(l+r)/2;
	build(id*2,l,mid);build(id*2+1,mid+1,r);
	tr[id].sum=tr[id*2].sum+tr[id*2+1].sum;
}
void down(int id){
	tr[id*2].add+=tr[id].add;
	tr[id*2].sum+=(tr[id*2].r-tr[id*2].l+1)*tr[id].add;
	tr[id*2+1].add+=tr[id].add;
	tr[id*2+1].sum+=(tr[id*2+1].r-tr[id*2+1].l+1)*tr[id].add;
	tr[id].add=0;
}
void update(int id,int l,int r,int v){
	if(tr[id].r<tr[id].l || tr[id].l>tr[id].r) return;
	if(tr[id].r<=r && tr[id].l>=l){
		tr[id].sum+=(tr[id].r-tr[id].l+1)*v;
		tr[id].add=v;
		return ;
	}
	if(tr[id].add!=0) down(id);
	update(id*2,l,r,v);update(id*2+1,l,r,v);
	tr[id].sum=tr[id*2].sum+tr[id*2+1].sum;
}
int select(int id,int l,int r){
	if(tr[id].l>r || tr[id].r<l) return 0;
	if(tr[id].l>=l && tr[id].r<=r) return tr[id].sum;
	if(tr[id].add!=0) down(id);
	return tr[id*2].sum+tr[id*2+1].sum;
}
void change(int x,int y,int v){
	while(top[x]!=top[y]){
		if(dep[top[x]]>dep[top[y]]) swap(x,y);
		update(1,pos[top[x]],pos[x],v);
		x=f[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	update(1,pos[x]+1,pos[y],v);
}
int ask(int x,int y){
	int ans=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]>dep[top[y]]) swap(x,y);
		ans+=select(1,pos[top[x]],pos[x]);
		x=f[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	ans+=select(1,pos[x]+1,pos[y]);
	return ans;
}
signed main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n-1;i++){
		int u,v;
		cin>>u>>v;
		add(u,v,0);
		add(v,u,0);
	}
	dfs1(1,1,1);
	k=0;
	cnt=0;
	dfs2(1,1);
	build(1,1,n);
	for(int i=1;i<=m;i++){
		string opt;
		cin>>opt;
		if(opt=="P"){
			int u,v;
			cin>>u>>v;
			change(u,v,1);
		}
		if(opt=="Q"){
			int u,v;
			cin>>u>>v;
			cout<<ask(u,v)<<endl;
		}
	}
	return 0;
}

哪位大佬帮我看一下这个树剖为啥死循环???(P3038)

2024/1/22 14:29
加载中...