大佬玄关求调,90tps,只WA了最后一个点
查看原帖
大佬玄关求调,90tps,只WA了最后一个点
560291
the_Dream_Chaser楼主2024/6/28 19:09

大佬玄关求调,90tps,只WA了最后一个点 代码如下

#include<bits/stdc++.h>
#define long long ll
const int MAXN=6e5+16;
using namespace std;
struct E{
	int u,v,nxt;
}e[MAXN];
struct tree{
	int l,r,w,siz,la;
}tr[MAXN];
int n,m,rot,Mod,cnt=0,a[MAXN],b[MAXN];
int head[MAXN];
int num=1;
void add(int x,int y){
	e[++num].u=x;
	e[num].v=y;
	e[num].nxt=head[x];
	head[x]=num;
}
int read(){
    char c=getchar();int x=0,la=1;
    while(c<'0'||c>'9'){if(c=='-')la=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0',c=getchar();}
    return x*la;
}
int dep[MAXN];//深度 
int fa[MAXN];//父节点 
int son[MAXN];//重儿子 
int siz[MAXN];//子树的大小
int top[MAXN];//属于的重链的深度最浅的点编号
int dfn[MAXN];//点u的dfs序
int dfs1(int now,int la,int dp){//dfs1处理掉 dep,fa,son,siz数组
    dep[now]=dp;
    fa[now]=la;
    siz[now]=1;
    int maxson=-1;
    for(int i=head[now];i!=-1;i=e[i].nxt){
        if(e[i].v==la) continue;
        siz[now]+=dfs1(e[i].v,now,dp+1);
        
		if(siz[e[i].v]>maxson) maxson=siz[e[i].v],son[now]=e[i].v;
    }
    return siz[now];
    //=siz[e[i].v];
}
void biu(int k,int ll,int rr){//建树 
	tr[k].l=ll;
	tr[k].r=rr;
	tr[k].siz=rr-ll+1;
	if(ll==rr){
		tr[k].w=a[ll];
		return ;
	}
	int mid=(ll+rr)>>1;
	biu(k<<1,ll,mid);
	biu(k<<1|1,mid+1,rr);
	tr[k].w=(tr[k<<1].w+tr[k<<1|1].w)%Mod;
}
void dfs2(int now,int tp){//dfs1处理掉 dfn,top数组
	dfn[now]=++cnt;
	a[cnt]=b[now];
	top[now]=tp;
	if(!son[now])return ;
	dfs2(son[now],tp);
	for(int i=head[now];i!=-1;i=e[i].nxt){
		if(!dfn[e[i].v]){
			dfs2(e[i].v,e[i].v);
		}
	}
}
void pushdown(int k){//传标记 
	if(!tr[k].la) return ;
    tr[k<<1].w=(tr[k<<1].w+tr[k<<1].siz*tr[k].la)%Mod;
    tr[k<<1|1].w=(tr[k<<1|1].w+tr[k<<1|1].siz*tr[k].la)%Mod;
    tr[k<<1].la=(tr[k<<1].la+tr[k].la)%Mod;
    tr[k<<1|1].la=(tr[k<<1|1].la+tr[k].la)%Mod;
    tr[k].la=0;
}
void change(int k,int ll,int rr,int val){//区间修改 
	if(ll<=tr[k].l&&tr[k].r<=rr){
        tr[k].w+=tr[k].siz*val;
        tr[k].la+=val;
        return ;
    }
    pushdown(k);
    int mid=(tr[k].l+tr[k].r)>>1;
    if(ll<=mid)change(k<<1,ll,rr,val);
    if(rr>mid)change(k<<1|1,ll,rr,val);
    tr[k].w=(tr[k<<1].w+tr[k<<1|1].w)%Mod;
}
int ask(int k,int ll,int rr){//区间求和 
    int ans=0;
    if(ll<=tr[k].l&&tr[k].r<=rr)
        return tr[k].w;
    pushdown(k);
    int mid=(tr[k].l+tr[k].r)>>1;
    if(ll<=mid) ans=(ans+ask(k<<1,ll,rr))%Mod;
    if(rr>mid)  ans=(ans+ask(k<<1|1,ll,rr))%Mod;
    return ans;
}
void sum(int x,int y){//路径和
    int ans=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ans=(ans+ask(1,dfn[top[x]],dfn[x]))%Mod;
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    ans=(ans+ask(1,dfn[x],dfn[y]))%Mod;
    printf("%d\n",ans);
}
void Add(int x,int y,int val){//路径加值
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        change(1,dfn[top[x]],dfn[x],val);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])    swap(x,y);
    change(1,dfn[x],dfn[y],val);
}
int main(){
	memset(head,-1,sizeof(head));
	n=read();m=read();rot=read();Mod=read();
	for(int w,i=1;i<=n;i++)b[i]=read();
	for(int u,v,i=1;i<=n-1;i++){
		u=read(),v=read();
		add(u,v);
		add(v,u);
	}
	dfs1(rot,0,1);
	dfs2(rot,rot);
	biu(1,1,n);
	for(int i=1,op,x,y,z;i<=m;i++){
		cin>>op;
	 	if(op==1){
	 		cin>>x>>y>>z;
	 		z=z%Mod;
	 		Add(x,y,z);
		}
		if(op==2){
			cin>>x>>y;
			sum(x,y);
		}
		if(op==3){
			cin>>x>>z;
			change(1,dfn[x],dfn[x]+siz[x]-1,z%Mod);
		}
		if(op==4){
			cin>>x;
			int ans=ask(1,dfn[x],dfn[x]+siz[x]-1);
			cout<<ans<<endl;
		}
	}
} 
/*
5 5 2 24
7 3 7 8 0 
1 2
1 5
3 1
4 1
3 4 2
3 2 2
4 5
1 5 1 3
2 1 3


*/
2024/6/28 19:09
加载中...