SP6779 GSS7 - Can you answer these queries VII

Posted 2022-03-22 algebra-hy

纯数据结构题，没有思维难度。直接用线段树求最大子段和的方法完成树上路径的合并。注意链上合并顺序要符合序列的前后顺序。

#include <cstdio>
#include <cstring>
#define cd w<<1
const int S=200030;
int n,Q,h[S],v[S],nx[S],d[S],e[S],s[S],eg=1,fa[S],t[S],f[S],_=0,rk[S],a[S],dep[S];
inline int ma(int a,int b)return a>b?a:b;
inline int mi(int a,int b)return a>b?b:a;
inline void swap(int &a,int &b)a^=b^=a^=b;
inline void egadd(int uu,int vv)

    nx[++eg]=h[uu];h[uu]=eg;
    v[eg]=vv;

struct node

    int a,ls,rs,s,tag;
    
;
inline node merge(node b,node c)

    node a;
    a.ls=ma(b.ls,b.s+c.ls);
    a.rs=ma(c.rs,c.s+b.rs);
    a.s=b.s+c.s;
    a.a=ma(b.a,c.a);
    a.a=ma(ma(a.ls,a.rs),ma(b.rs+c.ls,a.a));
    a.a=ma(a.a,0);
    a.ls=ma(a.ls,0);
    a.rs=ma(a.rs,0);
    return a;
   
const int inf=(1<<30)-1;
struct Sgtr

    node sg[S<<2];
    inline int pos(int l,int r)return (l+r)|(l!=r);
    void build(int l,int r)
    
        int w=pos(l,r);
        // printf("Build (%d,%d)=%d\n",l,r,w);
        if (l==r)
        
            int x=a[rk[l]];
            // printf("--build (%d %d)\n",rk[l],x);
            if (x>0)
                sg[w]=(node)x,x,x,x,inf;
            else
                sg[w]=(node)0,0,0,x,inf;
            return;
        
        int mid=(l+r)>>1,lc=pos(l,mid),rc=pos(mid+1,r);
        build(l,mid);
        build(mid+1,r);
        sg[w]=merge(sg[lc],sg[rc]);
        sg[w].tag=inf;
    
    void init()
    
        memset(sg,0,sizeof(sg));
        build(1,n);
    
    inline void pushdown(int w,int l,int mid,int r)
    
        int lc=pos(l,mid),rc=pos(mid+1,r),x=sg[w].tag;
        if (x!=inf)
        
            sg[lc].tag=sg[rc].tag=x;
            if (x>=0)
            
                sg[lc]=(node)(mid-l+1)*x,(mid-l+1)*x,(mid-l+1)*x,(mid-l+1)*x,x;
                sg[rc]=(node)(r-mid)*x,(r-mid)*x,(r-mid)*x,(r-mid)*x,x;
            
            else
            
                sg[lc]=(node)0,0,0,(mid-l+1)*x,x;
                sg[rc]=(node)0,0,0,(r-mid)*x,x;
            
        
        sg[w].tag=inf;
    
    void modify(int l,int r,int ll,int rr,int x)
    
        int w=pos(l,r);
        // printf("MODIFY %d %d %d %d %d\n",l,r,ll,rr,x);
        if (ll<=l && r<=rr)
        
            if (x>=0)
            
                sg[w].a=sg[w].ls=sg[w].rs=sg[w].s=(r-l+1)*x;
                sg[w].tag=x;
            
            else
            
                sg[w].a=sg[w].ls=sg[w].rs=0;
                sg[w].s=(r-l+1)*x;
                sg[w].tag=x;
            
            return;
        
        int mid=(l+r)>>1,lc=pos(l,mid),rc=pos(mid+1,r);
        pushdown(w,l,mid,r);
        if (ll<=mid) modify(l,mid,ll,rr,x);
        if (rr>mid) modify(mid+1,r,ll,rr,x);
        sg[w]=merge(sg[lc],sg[rc]);
        sg[w].tag=inf;
    
    node calcu(int l,int r,int ll,int rr)
    
        int w=pos(l,r);
        // printf("(%d %d)=%d  %d %d\n",l,r,w,ll,rr);
        if (ll<=l && r<=rr)
        
            // printf("node calcu(%d,%d) = %d %d %d %d %d\n",l,r,sg[w].a,sg[w].ls,sg[w].rs,sg[w].s,sg[w].tag);
            return sg[w];
        
        int mid=(l+r)>>1;
        pushdown(w,l,mid,r);
        if (ll>mid) return calcu(mid+1,r,ll,rr);
        if (rr<=mid) return calcu(l,mid,ll,rr);
        node a,b,c;
        b=calcu(l,mid,ll,rr);
        c=calcu(mid+1,r,ll,rr);
        a=merge(b,c);
        return a;
    /*
    void travel(int l,int r)
    
        int w=pos(l,r),mid=(l+r)>>1;
        printf("TRavel (%d,%d) = %d %d %d %d %d\n",l,r,sg[w].a,sg[w].ls,sg[w].rs,sg[w].s,sg[w].tag);
        if (l==r) return;
        travel(l,mid);
        travel(mid+1,r);
    */
sg;
void dfs_1(int x)

    s[x]=1;
    for (int i=h[x];i;i=nx[i])
        if (v[i]!=fa[x])
        
            fa[v[i]]=x;
            dep[v[i]]=dep[x]+1;
            dfs_1(v[i]);
            s[x]+=s[v[i]];
            if (s[v[i]]>s[e[x]])
                e[x]=v[i];
        

void dfs_2(int x,int top)

    f[x]=++_;
    rk[_]=x;
    t[x]=top;
    if (!e[x]) return;
    dfs_2(e[x],top);
    for (int i=h[x];i;i=nx[i])
        if (v[i]!=fa[x] && v[i]!=e[x])
            dfs_2(v[i],v[i]);

void _modify(int x,int y,int o)

    while (t[x]!=t[y])
    
        if (dep[t[x]]<dep[t[y]])
            swap(x,y);
        sg.modify(1,n,f[t[x]],f[x],o);
        // printf("modify %d~%d\n",t[x],x);
        x=fa[t[x]];
    
    if (dep[x]>dep[y])
        swap(x,y);
    // printf("modify %d~%d\n",x,y);
    sg.modify(1,n,f[x],f[y],o);

int _calcu(int x,int y)

    node hida,migi,ans;
    hida=migi=(node)0,0,0,0,inf;
    while (t[x]!=t[y])
    
        if (dep[t[x]]>dep[t[y]])
        
            // printf("Calcu: x:%d~%d   value=%d \n",f[t[x]],f[x],hida.a);
            // printf("o=%d ",o.a);
            hida=merge(sg.calcu(1,n,f[t[x]],f[x]),hida);
            // printf(" v2=%d\n",hida.a);
            x=fa[t[x]];
        
        else
        
            migi=merge(sg.calcu(1,n,f[t[y]],f[y]),migi);
            // printf("Calcu: y:%d~%d\n",t[y],y);
            y=fa[t[y]];
        
    
    if (dep[x]<dep[y])
    
        migi=merge(sg.calcu(1,n,f[x],f[y]),migi);
        y=x;
    
    else
    
        hida=merge(sg.calcu(1,n,f[y],f[x]),hida);
        x=y;
    
    // printf("Lca=%d\n",x);
    swap(hida.ls,hida.rs);
    ans=merge(hida,migi);
    return ans.a;

int main()

    scanf("%d",&n);
    for (int i=1;i<=n;i++)
        scanf("%d",a+i);
    for (int i=1,uu,vv;i<n;i++)
    
        scanf("%d%d",&uu,&vv);
        egadd(uu,vv);
        egadd(vv,uu);
    
    dfs_1(1);
    dfs_2(1,1);
    sg.init();
    // for (int i=1;i<=n;i++)
        // printf("(f=%d,%d,%d) ",f[i],a[i],sg.sg[sg.pos(f[i],f[i])].a);puts("");
    for (scanf("%d",&Q);Q--;)
    
        int o,x,y,z;
        scanf("%d%d%d",&o,&x,&y);
        if (o==1)
            printf("%d\n",_calcu(x,y));
        else
        
            scanf("%d",&z);
            _modify(x,y,z);
        
        // puts("---------------");sg.travel(1,n);