蒟蒻求调,样例都没过o(╥﹏╥)o
查看原帖
蒟蒻求调,样例都没过o(╥﹏╥)o
365532
Mr_ll楼主2021/10/6 21:16

样例输出2 21,我2 18

谢谢各位大佬了,拜托啦

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5;
int n,m,r,p,ysz[N+5],h[N+5],cnt,x,y,dep[N+5],son[N+5],siz[N+5],id[N+5],top[N+5],fa[N+5],idw[N+5],sum[N*2+5],op,z,laz[N*2+5];
struct qwe{
	int to,net;
}tr[N];
void add(int x,int y){
	tr[++cnt].to=y;
	tr[cnt].net=h[x];
	h[x]=cnt;
}
void push_up(int t){
	sum[t]=sum[t<<1]+sum[t<<1|1]; 
	sum[t]%=p;
}
void updown(int t,int l,int r){
	int mid=(l+r)>>1;
	laz[t<<1]+=laz[t];sum[t<<1]+=(mid-l+1)*laz[t];sum[t<<1]%=p;
	laz[t<<1|1]+=laz[t];sum[t<<1|1]+=(r-mid)*laz[t];sum[t<<1|1]%=p;
	laz[t]=0; 
}
void biu(int t,int l,int r){
	if(l==r){
		sum[t]=idw[l];
		return;
	}
	int mid=(l+r)>>1;
	biu(t<<1,l,mid);
	biu(t<<1|1,mid+1,r);
	push_up(t);
}
void add(int t,int l,int r,int x,int y,int k){
	if(l>=x&&r<=y){
		laz[t]+=k;
		sum[t]+=(r-l+1)*k; 
		sum[t]%=p;
		return;
	}
	int mid=(l+r)>>1;
	updown(t,l,r);
	if(mid>=x) add(t<<1,l,mid,x,y,k);
	if(mid<y) add(t<<1|1,mid+1,r,x,y,k);
	push_up(t);
}
int cha(int t,int l,int r,int x,int y){
	if(l>=x&&r<=y) return sum[t];
	int mid=(l+r)>>1;
	updown(t,l,r);
	int ans=0;
	if(mid>=x) ans+=cha(t<<1,l,mid,x,y);ans%=p;
	if(mid<y) ans+=cha(t<<1|1,mid+1,r,x,y);ans%=p;
	return ans;
}
void dfs1(int t,int f,int de){
	dep[t]=de;
	siz[t]=1;
	fa[t]=f;
	int max_son=-1;
	for(int i=h[t];i;i=tr[i].net){
		if(tr[i].to!=f){
			dfs1(tr[i].to,t,de+1);
			if(siz[tr[i].to]>max_son) son[t]=tr[i].to,max_son=siz[tr[i].to];
			siz[t]+=siz[tr[i].to]; 
		}
	}
}
void dfs2(int t,int topf){
	id[t]=++cnt;
	top[t]=topf;
	idw[t]=ysz[t];
	if(!son[t]) return;
	dfs2(son[t],topf);
	for(int i=h[t];i;i=tr[i].net){
		int too=tr[i].to;
		if(too!=fa[t]&&too!=son[t])
		dfs2(too,too);
	}
}
void sadd1(int x,int y,int z){
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		add(1,1,n,id[top[x]],id[x],z);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	add(1,1,n,id[x],id[y],z);
}
int scha1(int x,int y){
	int ans;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		ans+=cha(1,1,n,id[top[x]],id[x]);ans%=p;
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	ans+=cha(1,1,n,id[x],id[y]);ans%=p;
	return ans;
}
void sadd2(int x,int k){
	add(1,1,n,id[x],id[x]+siz[x]-1,k);
}
int scha2(int x){
	int ans=0;
	ans=cha(1,1,n,id[x],id[x]+siz[x]-1);
	ans%=p;
	return ans;
}
int main(){
	scanf("%d%d%d%d",&n,&m,&r,&p);
	for(int i=1;i<=n;i++) scanf("%d",&ysz[i]);
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	} 
	cnt=0;
	dfs1(r,0,1);
	dfs2(r,r);
	biu(1,1,n);
	for(int i=1;i<=m;i++){
		scanf("%d",&op);
		if(op==1){
			scanf("%d%d%d",&x,&y,&z);
			sadd1(x,y,z);
		}
		if(op==2){
			scanf("%d%d",&x,&y);
			printf("%d\n",scha1(x,y));
		}
		if(op==3){
			scanf("%d%d",&x,&z);
			sadd2(x,z);
		}
		if(op==4){
			scanf("%d",&x);
			printf("%d\n",scha2(x));
		}
	}
	return 0;
}
2021/10/6 21:16
加载中...