SP6779 GSS7 - Can you answer these queries VII

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

#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)
        {
            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;
    }
}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);
    }
}