題目描述

一棵nn個點的樹，每個點的初始權值為11。
對於這棵樹有qq個操作，每個操作為以下四種操作之一：

• + u v c：將uu到vv的路徑上的點的權值都加上自然數cc；

• - u1 v1 u2 v2：將樹中原有的邊(u_1,v_1)(u1​,v1​)刪除，加入一條新邊(u_2,v_2)(u2​,v2​)，保證操作完之後仍然是一棵樹；

• * u v c：將uu到vv的路徑上的點的權值都乘上自然數cc；

• / u v：詢問uu到vv的路徑上的點的權值和，將答案對5106151061取模。

輸入格式

第一行兩個整數n,qn,q

接下來n-1n−1行每行兩個正整數u,vu,v，描述這棵樹的每條邊。

接下來qq行，每行描述一個操作。

輸出格式

對於每個詢問操作，輸出一行一個整數表示答案。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+100;
const int mod=51061;
typedef long long ll;
ll f[maxn];
ll ch[maxn][2];
ll v[maxn];
ll size[maxn];
ll lm[maxn];
ll la[maxn];
ll s[maxn];
ll st[maxn];
bool r[maxn];
bool nroot (int
 
 x) {
    return ch[f[x]][0]==x||ch[f[x]][1]==x;
}
void pushup (int x) {
    s[x]=s[ch[x][0]]+s[ch[x][1]]+v[x];
    s[x]%=mod;
    
    size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
}

void rev (int x) {
    //翻轉
    int t=ch[x][0];
    ch[x][0]=ch[x][1];
    ch[x][1]=t;
    r[x]^=1; 
}

void push_mul (int x,int
 
 c) {
    //乘
    s[x]*=c;s[x]%=mod;
    v[x]*=c;v[x]%=mod;
    lm[x]*=c;lm[x]%=mod;
    la[x]*=c;la[x]%=mod;
}

void push_add (int x,int c) {
    //加 
    s[x]+=c*size[x];s[x]%=mod;
    v[x]+=c;v[x]%=mod;
    la[x]+=c;la[x]%=mod;
}

void pushdown (int x) {
    if (lm[x]!=1) push_mul(ch[x][0],lm[x]),push_mul(ch[x][1],lm[x]),lm[x]=1;
    if (la[x]) push_add(ch[x][0],la[x]),push_add(ch[x][1],la[x]),la[x]=0;
    if (r[x]) {
        if (ch[x][0]) rev(ch[x][0]);
        if (ch[x][1]) rev(ch[x][1]);
        r[x]=0;
    } 
}

void rotate (int x) {
    int y=f[x],z=f[y],k=ch[y][1]==x,w=ch[x][!k];
    if (nroot(y)) ch[z][ch[z][1]==y]=x;
    ch[x][!k]=y;
    ch[y][k]=w;
    if (w) f[w]=y;
    f[y]=x;f[x]=z;
    pushup(y);
}

void splay (int x) {
    int y=x,z=0;
    st[++z]=y;
    while (nroot(y)) st[++z]=y=f[y];
    while (z) pushdown(st[z--]);
    while (nroot(x)) {
        y=f[x];z=f[y];
        if (nroot(y)) rotate((ch[y][0]==x)^(ch[z][0]==y)?x:y);
        rotate(x);
    }
    pushup(x);
}

void access (int x) {
    for (int y=0;x;x=f[y=x]) splay(x),ch[x][1]=y,pushup(x);
}

void makeroot(int x) {
    access(x);
    splay(x);
    rev(x);
}

void split (int x,int y) {
    makeroot(x);
    access(y);
    splay(y);
}

void link (int x,int y) {
    makeroot(x);
    f[x]=y;
}

void cut (int x,int y) {
    split(x,y);
    f[x]=ch[y][0]=0;
}
int n,q;
int main () {
    scanf("%d%d",&n,&q); 
    for (int i=1;i<=n;i++) v[i]=size[i]=lm[i]=1;
    for (int i=1;i<n;i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        link(x,y);
    }
    while (q--) {
        string op;
        cin>>op;
        if (op=="+") {
            int a,b,k;
            scanf("%d%d%d",&a,&b,&k);
            split(a,b);
            push_add(b,k);
        }
        else if (op=="-") {
            int a,b;
            scanf("%d%d",&a,&b);cut(a,b);
            scanf("%d%d",&a,&b);link(a,b);
        }
        else if (op=="*") {
            int a,b,k;
            scanf("%d%d%d",&a,&b,&k);
            split(a,b);
            push_mul(b,k);
        }
        else if (op=="/") {
            int a,b;
            scanf("%d%d",&a,&b);
            split(a,b);
            printf("%lld\n",s[b]%mod);
        }
    }
}