JZOJ 5662 尺树寸泓

Description

给出一棵有$n$个节点的二叉树，每个点都有一个点值$w_i$，定义一个节点的力量值为以该点为根的子树内的所有点的点值之和，接下来$q$个曹作，分两种一个询问曹作，询问一个节点子树内的所有节点的力量值的积模$10^9+7$，或是旋转曹作，左旋/佑旋某个节点，如果旋转不了，请忽略这条曹作。

Data Constraint

$n,q<=10^5$ $0<w_i<10^9+7$

Solution

不难发现一次左旋/佑旋曹作只会改变两个节点的力量值并且左旋/佑旋曹作不会改变原二叉树的中序遍历，用一颗线段树来维护中序遍历在某段区间内的点的力量值之积，并且在进行左旋/佑旋曹作时同时维护一下节点子树内的点的中序遍历的最小值和最大值即可。

时间复杂度$O(q\ log\ n)$

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)

using namespace std;
typedef long long ll;
const ll N=32e4,mo=1e9+7,M=N<<2;

int fa[N],ls[N],rs[N],dfn[N],dy[N],be[N],en[N];
ll size[N],w[N];
int n,q,pp;
ll tr[M],ans;

inline int read()
{
    int o=0; char ch=' ';
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';ch=getchar())o=o*10+ch-48;
    return o;
}

void cor(int o,int l,int r,int op)
{
    if(l==r){
        tr[o]=size[dy[l]];
        return;
    }
    int mid=l+r>>1;
    if(op<=mid)cor(o<<1,l,mid,op);
    else cor((o<<1)^1,mid+1,r,op);
    tr[o]=tr[o<<1]*tr[(o<<1)^1]%mo;
}

void ask(int o,int l,int r,int le,int ri)
{
    if(l==le&&r==ri){
        ans=ans*tr[o]%mo;
        return;
    }
    int mid=l+r>>1;
    if(ri<=mid)ask(o<<1,l,mid,le,ri);
    else if(le>mid)ask((o<<1)^1,mid+1,r,le,ri);
    else{
        ask(o<<1,l,mid,le,mid);
        ask((o<<1)^1,mid+1,r,mid+1,ri);
    }
}

void dg(int o)
{
    if(ls[o])dg(ls[o]),be[o]=be[ls[o]],fa[ls[o]]=o;else be[o]=pp+1;
    dfn[o]=++pp; dy[pp]=o; en[o]=pp;
    if(rs[o])dg(rs[o]),en[o]=en[rs[o]],fa[rs[o]]=o;
    size[o]=(size[ls[o]]+size[rs[o]]+w[o])%mo;
    cor(1,1,n,dfn[o]);
}

inline void left_Rotate(int o)
{
    if(rs[o]==0)return;
    int ff=fa[o],yb=rs[o];
    fa[yb]=ff; rs[o]=ls[yb]; ls[yb]=o;
    if(ls[ff]==o)ls[ff]=yb;else rs[ff]=yb;
    fa[o]=yb; fa[rs[o]]=o;
    size[yb]=size[o];
    size[o]=(size[ls[o]]+size[rs[o]]+w[o])%mo;
    cor(1,1,n,dfn[yb]); cor(1,1,n,dfn[o]);
    if(ls[o])be[o]=be[ls[o]];else be[o]=dfn[o];
    if(rs[o])en[o]=en[rs[o]];else en[o]=dfn[o]; 
    if(ls[yb])be[yb]=be[ls[yb]];else be[yb]=dfn[yb];
    if(rs[yb])en[yb]=en[rs[yb]];else en[yb]=dfn[yb]; 
}

inline void right_Rotate(int o)
{
    if(ls[o]==0)return;
    int ff=fa[o],zb=ls[o];
    fa[zb]=ff; ls[o]=rs[zb]; rs[zb]=o;
    if(ls[ff]==o)ls[ff]=zb;else rs[ff]=zb;
    fa[o]=zb; fa[ls[o]]=o;
    size[zb]=size[o];
    size[o]=(size[ls[o]]+size[rs[o]]+w[o])%mo;
    cor(1,1,n,dfn[zb]); cor(1,1,n,dfn[o]);
    if(ls[o])be[o]=be[ls[o]];else be[o]=dfn[o];
    if(rs[o])en[o]=en[rs[o]];else en[o]=dfn[o]; 
    if(ls[zb])be[zb]=be[ls[zb]];else be[zb]=dfn[zb];
    if(rs[zb])en[zb]=en[rs[zb]];else en[zb]=dfn[zb]; 
}

int main()
{
    cin>>n>>q;
    fo(i,1,n){
        w[i]=read(); ls[i]=read(); rs[i]=read();
        if(ls[i])fa[ls[i]]=i;
        if(rs[i])fa[rs[i]]=i; 
    }
    fa[1]=0; dg(1);
    fo(i,1,q){
        int opt=read(),x=read();
        if(opt==2){
            ans=1;
            ask(1,1,n,be[x],en[x]);
            printf("%lld\n",ans);
        }
        if(opt==1)left_Rotate(x);
        if(opt==0)right_Rotate(x);
    }
}