求助,30分,调了一下午了。。
查看原帖
求助,30分,调了一下午了。。
272314
Autonomier楼主2020/7/28 17:35
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+10;
inline int read()
{
    int s=0,w=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
    return s*w;
}
int n,m,rt,mod;
int r[maxn],dep[maxn],fa[maxn],top[maxn],son[maxn],siz[maxn],id[maxn],w[maxn];
int head[maxn];
struct Eage
{
    int to;
    int nt;
}e[maxn*2];
int cnt;
inline void add_eage(int x,int y)
{
    e[++cnt].nt=head[x];
    e[cnt].to=y;
    head[x]=cnt;
}
inline void dfs1(int rt,int d,int pre)
{
    int max_siz=0;
    fa[rt]=pre;
    dep[rt]=d+1;
    siz[rt]=1;
    for(int i=head[rt];i;i=e[i].nt)
    {
        int y=e[i].to;
        if(fa[y])
            continue;
        dfs1(y,dep[rt],rt);
        siz[rt]+=siz[y];
        if(siz[y]>max_siz)
        {
            max_siz=siz[y];
            son[rt]=y;
        }   
    }   
}
int dfn;
inline void dfs2(int rt,int ld)
{
    id[rt]=++dfn;
    top[rt]=ld;
    w[dfn]=r[rt];
    if(!son[rt])
        return;
    dfs2(son[rt],ld);
    for(int i=head[rt];i;i=e[i].nt)
    {
        int y=e[i].to;
        if(y==fa[rt]||y==son[rt])
            continue;
        dfs2(y,y);
    }
}
int t[maxn*4],tag[maxn*4];
inline void push_up(int p)
{
    t[p]=t[p<<1|1]+t[p<<1];
}
inline void push_down(int p,int l,int r)
{
    tag[p<<1|1]+=tag[p];
    tag[p<<1]+=tag[p];
	int mid=(l+r)>>1;
	t[p<<1]+=tag[p]*(mid-l+1)%mod;
	t[p<<1]%=mod;
	t[p<<1|1]+=tag[p]*(r-mid)%mod;
	t[p<<1|1]%=mod;
    tag[p]=0;
}
inline void build(int p,int l,int r)
{
    if(l==r)
    {
        t[p]=w[l]%mod;
        return;
    }
    int mid=(l+r)>>1;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
    push_up(p);
}
inline void update(int p,int l,int r,int ul,int ur,int x)
{
    if(ul<=l&&ur>=r)
    {
        tag[p]+=x%mod;
        t[p]+=(r-l+1)*x;
        t[p]%=mod;
        return;
    }
    if(tag[p])
		push_down(p,l,r);
    int mid=(l+r)>>1;
    if(ul<=mid)
        update(p<<1,l,mid,ul,ur,x);
    if(ur>mid)
        update(p<<1|1,mid+1,r,ul,ur,x);
    push_up(p);
}
inline int query(int p,int l,int r,int ql,int qr)
{
	int res=0;
    if(l>=ql&&r<=qr)
    {
        return t[p]%mod;
    }
    if(tag[p])
		push_down(p,l,r);
    int mid=(l+r)>>1;
    if(ql<=mid)
        res+=query(p<<1,l,mid,ql,qr)%mod;
    if(qr>mid)
        res+=query(p<<1|1,mid+1,r,ql,qr)%mod;
    return res%mod;
}
int main()
{
    n=read(),m=read(),rt=read(),mod=read();
    for(int i=1;i<=n;i++)
    {
        r[i]=read();
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        x=read(),y=read();
        add_eage(x,y);
        add_eage(y,x);
    }
    dfs1(rt,0,rt);
    dfs2(rt,rt);
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int opt,x,y,z;
        opt=read();
        if(opt==1)   
        {
            x=read(),y=read(),z=read();
            while(top[x]!=top[y])
            {
                if(dep[top[x]]<dep[top[y]])
                    swap(x,y);
                update(1,1,n,id[top[x]],id[x],z);
                x=fa[top[x]];
            }
            if(dep[x]<dep[y])
                swap(x,y);
            update(1,1,n,id[y],id[x],z);
        }
        else if(opt==2)
        {
            x=read(),y=read();
            int res=0;
            while(top[x]!=top[y])
            {
                if(dep[top[x]]<dep[top[y]])
                    swap(x,y);
                res+=query(1,1,n,id[top[x]],id[x])%mod;
                x=fa[top[x]];
            }
            if(dep[x]<dep[y])
                swap(x,y);
            res+=query(1,1,n,id[y],id[x])%mod;
           	printf("%d\n",res);
        }
        else if(opt==3)
        {
            x=read(),y=read();
            update(1,1,n,id[x],id[x]+siz[x]-1,y);
        }
        else
        {
            x=read();
           	printf("%d\n",query(1,1,n,id[x],id[x]+siz[x]-1));
        }
    }
     return 0;
}

2020/7/28 17:35
加载中...