[国家集训队] tree Ⅱ

bzoj2631(权限题。。。):链接
落咕:链接
考察的是LCT维护链上信息。
但是这个题不一样的是又有加法又有乘法。。。(有木有想到落咕的模板——线段树2qwq)
因为乘法的运算优先度比加法高,所以我们要先做乘法再做加法(证明?去看线段树2的题解吧),push_down的时候要注意一下。
处理一条路径的时候,直接split拉出来,然后在根节点进行修改就行了,push_down的时候会把它的标记传递下去,也就完成了一条路径上的加乘操作。
注意计算乘法的时候别忘了*1ll,因为有可能爆int(好吧,作为惯例,一般涉及到乘法再取模操作的时候,中间都要加一个强制类型转换防止爆int qwqwq)
最后还有一个坑点,数据输入的时候有乘0的情况,所以我们在push_down的时候判断有没有乘法标记不能写if(t[x].mul>1),正确写法应该是if(t[x].mul!=1)。。。。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 100010
#define mod 51061
using namespace std;
int n,m,tot;
int s[MAXN];
struct Node{int add,mul,v,rev,son,sum,ff,ch[2];}t[MAXN];
inline void push_up(int x)
{
    t[x].sum=(t[t[x].ch[0]].sum+t[t[x].ch[1]].sum+t[x].v)%mod;
    t[x].son=t[t[x].ch[0]].son+t[t[x].ch[1]].son+1;
}
inline bool isroot(int x){return t[t[x].ff].ch[0]!=x&&t[t[x].ff].ch[1]!=x;}
inline void rotate(int x)
{
    int y=t[x].ff;
    int z=t[y].ff;
    int k=t[y].ch[1]==x;
    if(!isroot(y)) t[z].ch[t[z].ch[1]==y]=x; t[x].ff=z;
    t[y].ch[k]=t[x].ch[k^1]; t[t[x].ch[k^1]].ff=y;
    t[x].ch[k^1]=y; t[y].ff=x;
    push_up(y),push_up(x);
}
inline void Mul(int x,int k)
{
    t[x].mul=1ll*t[x].mul*k%mod;
    t[x].add=1ll*t[x].add*k%mod;
    t[x].v=1ll*t[x].v*k%mod;
    t[x].sum=1ll*t[x].sum*k%mod;
}
inline void Add(int x,int k)
{
    t[x].add=(t[x].add+k)%mod;
    t[x].v=(t[x].v+k)%mod;
    t[x].sum+=1ll*k*t[x].son%mod,t[x].sum%=mod;
}
inline void push_down(int x)
{
    if(t[x].rev)
    {
        if(t[x].ch[0]) t[t[x].ch[0]].rev^=1;
        if(t[x].ch[1]) t[t[x].ch[1]].rev^=1;
        swap(t[x].ch[0],t[x].ch[1]);
        t[x].rev^=1;
    }
    if(t[x].mul!=1)
    {
        if(t[x].ch[0]) Mul(t[x].ch[0],t[x].mul);
        if(t[x].ch[1]) Mul(t[x].ch[1],t[x].mul);
        t[x].mul=1;
    }
    if(t[x].add)
    {
        if(t[x].ch[0]) Add(t[x].ch[0],t[x].add);
        if(t[x].ch[1]) Add(t[x].ch[1],t[x].add);
        t[x].add=0;
    }
}
inline void splay(int x)
{
    s[tot=1]=x;
    for(int i=x;!isroot(i);i=t[i].ff) s[++tot]=t[i].ff;
    while(tot) push_down(s[tot--]);
    while(!isroot(x))
    {
        int y=t[x].ff;
        int z=t[y].ff;
        if(!isroot(y))
            ((t[y].ch[0]==x)^(t[z].ch[0]==y))?rotate(x):rotate(y);
        rotate(x);
    }
}
inline void access(int x)
{
    for(int y=0;x;y=x,x=t[x].ff)
        splay(x),t[x].ch[1]=y,push_up(x);
}
inline void makeroot(int x){access(x);splay(x);t[x].rev^=1;}
inline void split(int x,int y){makeroot(x);access(y);splay(y);}
inline void cut(int x,int y){split(x,y);t[y].ch[0]=t[x].ff=0;push_up(y);}
inline void link(int x,int y){makeroot(x);t[x].ff=y;}
inline int findroot(int x)
{   
    access(x);
    splay(x);
    while(t[x].ch[0]) x=t[x].ch[0];
    return x;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        link(u,v);
    }
    for(int i=1;i<=n;i++) t[i].son=t[i].mul=t[i].v=1;
    for(int i=1;i<=m;i++)
    {
        char op;
        int u,v,c,u2,v2;
        scanf("%c",&op),scanf("%c",&op);
        if(op=='+')
        {
            scanf("%d%d%d",&u,&v,&c);
            split(u,v);
            Add(v,c);
        }
        else if(op=='-')
        {
            scanf("%d%d%d%d",&u,&v,&u2,&v2);
            cut(u,v);
            link(u2,v2);
        }
        else if(op=='*')
        {
            scanf("%d%d%d",&u,&v,&c);
            split(u,v);
            Mul(v,c);
        }
        else
        {
            scanf("%d%d",&u,&v);
            split(u,v);
            printf("%d\n",t[v].sum%mod);
        }
    } 
    return 0;
} 

猜你喜欢

转载自www.cnblogs.com/fengxunling/p/10285777.html