萌新刚学树剖,70分求助QwQ

回复帖子

@_DXL_  2019-10-02 10:38 回复

代码如下:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=210000;
ll n,m,r,p,a[N],d[N],f[N],son[N],size[N],id[N],top[N],pro[N],nowid,leaf[N],flag[N];
ll head[N],maxE; struct Edge{ll nxt,to;}e[N];
inline ll read()
{
    ll s=0,w=1; char ch;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;
}
inline void Add(ll u,ll v) {e[++maxE].nxt=head[u];head[u]=maxE;e[maxE].to=v;}
void DFS_First(ll x,ll before)
{
    d[x]=d[before]+1;f[x]=before;size[x]=1;
    for(ll i=head[x];i;i=e[i].nxt)
    {
        if(e[i].to==before) continue;
        DFS_First(e[i].to,x);size[x]+=size[e[i].to];
        if(size[e[i].to]>size[son[x]]) son[x]=e[i].to;
    }
}
void DFS_Second(ll x,ll tk)
{
    id[x]=++nowid; pro[nowid]=a[x]; top[x]=tk;
    if(!son[x]) return; DFS_Second(son[x],tk);
    for(ll i=head[x];i;i=e[i].nxt) {if(e[i].to==f[x]||e[i].to==son[x]) continue; DFS_Second(e[i].to,e[i].to);} 
}
inline ll LeftChild(ll x) {return x<<1;}
inline ll RightChild(ll x) {return x<<1|1;}
inline void Push_Up(ll x) {leaf[x]=(leaf[LeftChild(x)]+leaf[RightChild(x)])%p;}
void Build_Tree(ll x,ll l,ll r)
{
    if(l==r) {leaf[x]=pro[l]; leaf[x]%=p; return;}
    Build_Tree(LeftChild(x),l,(l+r)>>1);
    Build_Tree(RightChild(x),((l+r)>>1)+1,r);
    Push_Up(x);
}
inline void Add_Up(ll x,ll l,ll r,ll k) {flag[x]=(flag[x]+k)%p; leaf[x]=(leaf[x]+k*(r-l+1))%p;}
inline void Push_Down(ll x,ll l,ll r)
{
    Add_Up(LeftChild(x),l,(l+r)>>1,flag[x]);
    Add_Up(RightChild(x),((l+r)>>1)+1,r,flag[x]);
    flag[x]=0;
}
void UpDate(ll u,ll v,ll l,ll r,ll x,ll k)
{
    if(l>=u&&r<=v) {Add_Up(x,l,r,k); return;}
    Push_Down(x,l,r);
    if(u<=(l+r)>>1) UpDate(u,v,l,(l+r)>>1,LeftChild(x),k);
    if(v>(l+r)>>1) UpDate(u,v,((l+r)>>1)+1,r,RightChild(x),k);
    Push_Up(x);
}
ll Ask(ll u,ll v,ll l,ll r,ll x)
{
    ll ans=0;
    if(l>=u&&r<=v) return leaf[x]%p;
    Push_Down(x,l,r);
    if(u<=(l+r)>>1) ans=(ans+Ask(u,v,l,(l+r)>>1,LeftChild(x)))%p;
    if(v>(l+r)>>1) ans=(ans+Ask(u,v,((l+r)>>1)+1,r,RightChild(x)))%p;
    return (ans+p)%p;
}
void UpDateRoad(ll l,ll r,ll k)
{
    k%=p;
    while(top[l]!=top[r])
    {
        if(d[top[l]]<d[top[r]]) swap(l,r);
        UpDate(id[top[l]],id[l],1,n,1,k);l=f[top[l]];
    }if(d[l]>d[r]) swap(l,r); UpDate(id[l],id[r],1,n,1,k);
}
ll AskRoad(ll l,ll r)
{
    ll ans=0;
    while(top[l]!=top[r])
    {
        if(d[top[l]]<d[top[r]]) swap(l,r);
        ans=(ans+Ask(id[top[l]],id[l],1,n,1))%p;
        l=f[top[l]];
    }if(d[l]>d[r]) swap(l,r); ans=(ans+Ask(id[l],id[r],1,n,1))%p;
    return (ans+p)%p;
}
int main()
{
    n=read();m=read();r=read();p=read();
    for(ll i=1;i<=n;i++) a[i]=read();
    for(ll i=1,u,v;i<n;i++) {u=read();v=read(); Add(u,v);Add(v,u);}
    DFS_First(r,0);DFS_Second(r,r);
    Build_Tree(1,1,n);
    for(ll i=1,op,x,y,z;i<=m;i++)
    {
        op=read();x=read();
        if(op==1) y=read(),z=read(),UpDateRoad(x,y,z);
        else if(op==2) y=read(),printf("%lld\n",AskRoad(x,y));
        else if(op==3) z=read(),UpDate(id[x],id[x]+size[x]-1,1,n,1,z);
        else if(op==4) printf("%lld\n",Ask(id[x],id[x]+size[x]-1,1,n,1));
    }
    return 0;
}
反馈
如果你认为某个帖子有问题,欢迎向洛谷反馈,以帮助更多的同学。



请具体说明理由,以增加反馈的可信度。