树剖模板求调,样例已过
查看原帖
树剖模板求调,样例已过
750803
_Revenge_楼主2022/8/7 21:32

RE+WA+MLE
我是看第一篇题解打的,思路一样,码风不同
调了一下午了呜呜呜

#include<bits/stdc++.h>

using namespace std;

typedef int ll;
const int N=3e5+50;
ll n,m,r,mod;

struct edge{
	ll to,next;
}e[N];
ll head[N],cnt;
void add_edge(int u,int v){
	++cnt;
	e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}

ll w[N],wt[N];
ll son[N],id[N],fa[N],dep[N],siz[N],top[N],res;

ll ans[N<<2],lazy_add[N<<2];

ll ls(ll p){return p<<1;}
ll rs(ll p){return p<<1|1;}
void push_up(ll p){
	ans[p]=ans[ls(p)]+ans[rs(p)]%mod;
}

void build(ll p,ll l,ll r){
	if(l==r){ans[p]=wt[l];ans[p]=ans[p]>mod?ans[p]%mod:ans[p];return;}
	ll mid=(l+r)>>1;
	build(ls(p),l,mid);
	build(rs(p),mid+1,r);
	push_up(p);
}

void change(ll p,ll l,ll r,ll add){
	lazy_add[p]=lazy_add[p]+add;
	ans[p]=(ans[p]+(r-l+1)*add)%mod;
}

void push_down(ll p,ll l,ll r){
	ll mid=(l+r)>>1;
	change(ls(p),l,mid,lazy_add[p]);
	change(rs(p),mid+1,r,lazy_add[p]);
	lazy_add[p]=0;
}

void update(ll nx,ll ny,ll l,ll r,ll p,ll k){
	if(nx<=l&&r<=ny){
		ans[p]+=(r-l+1)*k;
		lazy_add[p]+=k;
		return;
	}
	push_down(p,l,r);
	ll mid=(l+r)>>1;
	if(nx<=mid) update(nx,ny,l,mid,ls(p),k);
	if(ny>mid) update(nx,ny,mid+1,r,rs(p),k);
	push_up(p);
}

ll query(ll nx,ll ny,ll l,ll r,ll p){
	ll res=0;
	if(nx<=l&&r<=ny){return ans[p];}
	push_down(p,l,r);
	ll mid=(l+r)>>1;
	if(nx<=mid) res=res+query(nx,ny,l,mid,ls(p))%mod;
	if(ny>mid) res=res+query(nx,ny,mid+1,r,rs(p))%mod;
	return res;
}

void update_range(ll x,ll y,ll k){
	k%=mod;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		update(id[top[x]],id[x],1,n,1,k);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	update(id[x],id[y],1,n,1,k);
}

ll query_range(ll x,ll y){
	ll res=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		res=res+query(id[top[x]],id[x],1,n,1)%mod;
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	res=res+query(id[x],id[y],1,n,1)%mod;
	return res;
}

void update_tree(ll x,ll z){
	update(id[x],id[x]+siz[x]-1,1,n,1,z);
}

ll query_tree(ll x){
	return query(id[x],id[x]+siz[x]-1,1,n,1);
}

void dfs1(ll u,ll p){
	dep[u]=dep[p]+1;
	fa[u]=p;
	siz[u]=1;
	ll maxn=-1;
	for(int i=head[u];i;i=e[i].next){
		ll v=e[i].to;
		if(v==p) continue;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(siz[v]>maxn) son[u]=v,maxn=siz[v];
	}
}

void dfs2(ll u,ll topf){
	++res;
	id[u]=res;
	wt[res]=w[u];
	top[u]=topf;
	if(!son[u]) return;
	dfs2(son[u],topf);
	for(int i=head[u];i;i=e[i].next){
		ll v=e[i].to;
		if(v==fa[u]||v==son[u]) continue;
		dfs2(v,v);
	}
}

int main()
{
	scanf("%d%d%d%d",&n,&m,&r,&mod);
	for(int i=1;i<=m;++i) scanf("%d",&w[i]);
	for(int i=1;i<n;++i){
		ll u,v;
		scanf("%d%d",&u,&v);
		add_edge(u,v);
		add_edge(v,u);
	}
	dfs1(r,0);
	dfs2(r,r);
	build(1,1,n);
	for(int i=1;i<=m;++i){
		ll opt;
		scanf("%d",&opt);
		if(opt==1){
			ll x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			update_range(x,y,z);
		}else
		if(opt==2){
			ll x,y;
			scanf("%d%d",&x,&y);
			printf("%d\n",query_range(x,y));
		}else
		if(opt==3){
			ll x,z;
			scanf("%d%d",&x,&z);
			update_tree(x,z);
		}else{
			ll x;
			scanf("%d",&x);
			printf("%d\n",query_tree(x));
		}
	}
	return 0;
}
2022/8/7 21:32
加载中...