bzoj3159决战 码农题 树剖套splay

最近沉迷码农题无法自拔

首先有一个暴力的想法：对于每个重链维护一个splay，需要翻转的连起来，翻转，接回去

然后发现这样没问题。。。

一条链只能跨log个重链，也就只有log个splay的子树参与重排，所以一次翻转只要log^2的时间

需要维护的东西有点多

头一次在splay上维护这么多乱七八糟的东西，写错了好几遍

感觉学到不少debug技巧，也纠正了之前splay写法的一个小漏洞

话说极限数据好像和miaom拍不过啊。。。但是我们都A了啊

好烦啊不知道谁出锅了

#include <bits/stdc++.h>
#define num(x) (dep[x]-dep[top[x]]+1)
#define ans tr[split(num(x),num(y))]
#define rt root[RT]
#define RT1 root[n+1]
#define NO 800000
#define N 800000
#define INF 2000000000
#define ll long long
using namespace std;
struct trnode
{
    int size,min,max;ll sum;int plus,num;
    bool rev;
    trnode()
    {
        size=1;min=max=sum=plus=rev=num=0;
    }
} tr[NO];
int n,m,R,p,q,RT,x,y,z,RET,E,NODE,sttop,inv_x,inv_y;ll ret;
int c[NO][2],fa[NO];bool die[N];
int root[N],Fa[N],size[N],fir[N],top[N],dep[N],st[N],sz[N],lin[N];
int nex[2*N],to[2*N];
char cmd[10];
void up(int x)
{
    if(!x)
    {
        tr[0]=trnode();tr[0].size=0;
        return;
    }
    tr[x].size=tr[c[x][0]].size+tr[c[x][1]].size+1;
    tr[x].sum=tr[c[x][0]].sum+tr[c[x][1]].sum+(ll)tr[x].plus*tr[x].size+tr[x].num;
    tr[x].max=max(tr[c[x][0]].max,tr[c[x][1]].max)+tr[x].plus;
    tr[x].max=max(tr[x].max,tr[x].num+tr[x].plus);
    tr[x].min=tr[x].num;
    if(c[x][0]!=0)
        tr[x].min=min(tr[x].min,tr[c[x][0]].min);
    if(c[x][1]!=0)
        tr[x].min=min(tr[x].min,tr[c[x][1]].min);
    tr[x].min+=tr[x].plus;
}
void down(int x)
{
    if(tr[x].rev)
        swap(c[x][0],c[x][1]),tr[c[x][0]].rev^=1,tr[c[x][1]].rev^=1,tr[x].rev=0;
    if(tr[x].plus)
        tr[c[x][0]].plus+=tr[x].plus,tr[c[x][1]].plus+=tr[x].plus,
        tr[x].num+=tr[x].plus,tr[x].plus=0,up(c[x][0]),up(c[x][1]);
}
void rot(int x)
{
    down(fa[x]);down(x);
    int y=fa[x],k=c[y][1]==x;
    if(fa[y]) c[fa[y]][c[fa[y]][1]==y]=x;
    else rt=x;
    fa[x]=fa[y];fa[y]=x;
    c[y][k]=c[x][!k];
    if(c[x][!k])
        fa[c[x][!k]]=y;
    c[x][!k]=y;
    up(y);up(x);
}
void splay(int x,int z)
{
    for(int y;(y=fa[x])!=z;rot(x))
    if(fa[y]!=z)
        rot(c[fa[y]][1]==y^c[y][1]==x?x:y);
}
int Find(int x)
{
    if(x==0) return -1;
    if(x>tr[rt].size) return -1;
    int now=rt;
    while(down(now),tr[c[now][0]].size+1!=x)
    if(x<=tr[c[now][0]].size)
        now=c[now][0];
    else
        x-=tr[c[now][0]].size+1,now=c[now][1];
    return now;
}
int split(int x,int y)
{
    if(x==1 && y==tr[rt].size)
        return rt;
    if(x==1) 
    { splay(Find(y+1),0);return c[rt][0];}
    if(y==tr[rt].size) 
    { splay(Find(x-1),0);return c[rt][1];}
    splay(Find(x-1),0);
    splay(Find(y+1),rt);
    return c[c[rt][1]][0];
} 
void upd(int x)
{
    int len=0;
    while(x)
        lin[++len]=x,x=fa[x];
    for(int i=len;i;i--)
        down(lin[i]);
    for(int i=1;i<=len;i++)
        up(lin[i]);
}
int Split(int x,int y)
{
    int p=split(x,y),q=fa[p];
    upd(p);
    fa[p]=0;c[q][c[q][1]==p]=0;
    upd(q); 
    splay(q,0);
    return p;
}
void work(int x,int y)
{
    RT=top[x];int tem;
    if(cmd[2]=='c')
        tr[tem=split(num(x),num(y))].plus+=z,upd(tem);//?
    if(cmd[0]=='S')
        ret+=ans.sum;
    if(cmd[1]=='a')    ret=max(ret,(ll)ans.max);
    if(cmd[1]=='i')    RET=min(RET,ans.min);
    if(cmd[2]=='v')
    {
        int tem=Split(num(x),num(y));
        if(tem==rt) die[RT]=1;
        RT=n+1;
        if(!rt) rt=tem;
        else
        { 
            splay(Find(1),0);
            c[rt][0]=tem;fa[tem]=rt;
            splay(tem,0);
        }
    }
}
void add(int p,int q)
{
    to[++E]=q;nex[E]=fir[p];fir[p]=E;
}
int build(int now,int fat)
{
    size[now]=1;Fa[now]=fat;dep[now]=dep[fat]+1;
    for(int i=fir[now];i;i=nex[i])
    if(to[i]!=fat)
        size[now]+=build(to[i],now);
    return size[now];
}
void ins()
{
    int now=rt;
    while(c[now][1]) now=c[now][1];
    c[now][1]=++NODE;fa[NODE]=now;
    tr[NODE]=trnode();
    splay(NODE,0);
}
void pou(int now,int Top)
{
    top[now]=Top;
    if(Top==now)
        tr[++NODE]=trnode(),root[now]=NODE;
    else    
        RT=Top,ins();
    int id=0;
    for(int i=fir[now];i;i=nex[i])
    if(to[i]!=Fa[now])
    if(size[to[i]]>size[id])
        id=to[i];
    if(id) pou(id,Top);
    for(int i=fir[now];i;i=nex[i])
    if(to[i]!=Fa[now] && to[i]!=id)
        pou(to[i],to[i]);
}
int main()
{
    scanf("%d%d%d",&n,&m,&R);
    for(int i=1;i<n;i++)
        scanf("%d%d",&p,&q),
        add(p,q),add(q,p);
    tr[0].size=0; 
    build(R,0);
    pou(R,R);
    for(int i=1;i<=m;i++)
    {
        if(i==4)
            int e=1;
        scanf("%s%d%d",cmd,&x,&y);
        if(cmd[2]=='c') scanf("%d",&z); 
        ret=0;RET=INF;sttop=0;
        while(top[x]!=top[y])
        {
            if(dep[top[x]]<dep[top[y]])    swap(x,y);
            work(top[x],x);
            st[++sttop]=top[x];sz[sttop]=num(x);
            x=Fa[top[x]];
        }
        if(dep[x]>dep[y])
            swap(x,y);
        if(cmd[2]=='v')
        {
            RT=top[x];
            inv_x=Find(num(x)-1);inv_y=Find(num(y)+1);
        }
        work(x,y);
        if(cmd[0]=='S' || cmd[1]=='a') printf("%lld
",ret);
        if(cmd[1]=='i') printf("%d
",RET); 
        if(cmd[2]=='v')
        {
            RT=n+1;
            tr[rt].rev^=1;
            int tem=Split(1,num(y)-num(x)+1);
            RT=top[x];
            if(inv_x==-1 && inv_y==-1)
                rt=tem,die[RT]=0;
            else
            if(inv_x==-1)
            {
                splay(inv_y,0);
                c[rt][0]=tem;fa[tem]=rt;
                splay(tem,0);
            }
            else
            if(inv_y==-1)
                splay(inv_x,0),c[rt][1]=tem,fa[tem]=rt,splay(tem,0);
            else
                splay(inv_x,0),splay(inv_y,rt),c[c[rt][1]][0]=tem,fa[tem]=c[rt][1],splay(tem,0);
            for(int i=sttop;i;i--)
            {
                RT=n+1;
                int tem=Split(1,sz[i]);
                RT=st[i];
                if(die[RT]) die[RT]=0,rt=tem;
                else
                {
                    splay(Find(1),0);
                    c[rt][0]=tem;fa[tem]=rt;splay(tem,0);
                }
            }
        }
        root[n+1]=0;die[n+1]=0;
    }
    return 0;
}