#LOJ2564 SDOI2018 原题识别 主席树

转载请注明原文地址：http://www.cnblogs.com/LadyLex/p/9057297.html

原题链接：

今天考试考了前天的SDOI考题

天啊我菜爆，只有T2拿了30分

然后考试后半程一直在打T1

觉得考试思路很有意思，于是就顺着打下来了

个人感觉这个是$O(nlog^{2}n)$的，但是在loj上我比claris的程序快了1s多，只不过编程复杂度不止翻倍啊……

下面介绍一下我的解法

其实最早启发我的是链上的部分分

定义$pre_{i}$为i前面最近的和i同色的点的下标，我们把每个点的pre看成权值，插入一个主席树

这样的话，1操作我们只需要用主席树查$[L,R]$区间内pre<L的点数即可

然后我们考虑2操作，我的思路是统计每个点的贡献

对于这两个询问点，我们假设深度较浅的点为A，较深的为B

我们分B选的点在A前面还是后面来讨论

对于点i>A,那么点i的贡献应该是$[pre_{i}<=A]*(A-pre_{i})*(B-i+1)$

这个很好理解，前面布尔表达式是做贡献的前提，后面两个括号一个是左端点A的区间，一个是右端点B的区间

而对于i<=A，式子是类似的：贡献是$(i-pre[i])*(A-i+1+B-i+1)-1$

这里后面减1是因为A，B同时选择i点的情况被贡献了2次，所以要减一下

请选手手动把上面的括号都乘开……然后……

我们需要维护下面4个变量：$i$的和，$pre_{i}$的和，$i^{2}$的和，以及$i*pre_{i}$的和

由于空间有限，这里不展开写式子了

而当问题扩展到树上的时候……

我们先把刚才那个pre搬到树上来，

并且把pre的定义变成同色的最近祖先的深度

并且把编号全部改成深度

这两点非常重要，我在调试的时候因为有几处没有把原来序列上的编号改成深度调了半天……

然后把那个主席树也搬上来，维护从i到根的pre信息，这个两个操作都要用

再以颜色为下标建一个树上主席树，$root_{i}$维护i到根的颜色，这个我们1操作要用

我们发现这是个随机树

那么每个点距离链的深度应该会非常小

先考虑1操作，我们找到两个点的lca，这个可以通过直接暴力爬父亲直到走到链上解决，复杂度$O(logn)$

然后在维护pre的树上查询$pre<deep[lca]$的点的个数

再暴力爬深度较浅的那棵树，爬到lca，查询$lca---r$那条链里是否有这种颜色，以及这次暴力爬是否已经经过了这个元素

然后就得到了答案

这个其实我说的麻烦……可以参考一下代码，挺简单的

然后对于2操作呢……

我们还是分开考虑，具体来说我是这样搞得：

A选择$1-----lca$，B随意选择$1---r$，这样使用刚才链的算法可以统计

A选择$lca----l$，B选择$1----lca$，我们统计B选择的点和A选择的点，这个把刚才的式子稍微修改一下就能统计

具体来说，设$l----lca$的长度是L，那么A和B的贡献分别是

$(lca-pre_{i})*(deep[l]-deep[i]+1)$和

$(i-pre_{i})*l$

然后我们再考虑一种最难处理的情况：A在$lca---l$，B在$lca---r$

然后我们用和刚才类似的方法可以算B的贡献：$[pre_{i} < deep_{lca}]*(B-i+1)*l$

然后A的贡献的话，由于颜色也是随机的所以每种颜色个数比较少，我们可以暴力找到$lca---r$上第一个同色点然后算贡献

语言表述可能很让人难以理解……

请参考下面namespace sgt1和deal1&deal2函数来更好的理解上面的式子……

这样的话，我们的复杂度是$nlogn$*期望距链距离 + $nlogn$ + $n$*期望颜色个数*期望距链深度……比较玄学吧……

下面附上代码，保留了一定注释增强观赏体验……

#pragma GCC optimize("O3")
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
#define RG register
#define UI unsigned int
#define LL long long
#define N 100010
UI SA,SB,SC;
inline UI make()
{
    SA^=SA<<16;SA^=SA>>5;SA^=SA<<1;
    UI tmp=SA;
    SA=SB,SB=SC,SC^=SA^tmp;
    return SC;
}
int n,col[N],m,q,e,adj[N];
struct edge{int zhong,next;}s[N<<1];
inline void add(int qi,int zhong)
    {s[++e].zhong=zhong;s[e].next=adj[qi];adj[qi]=e;}
int fa[N],lst[N],pre[N],deep[N];
LL s1,s2,s3;int sz;
namespace sgt2
{
    struct node
    {
        node *ch[2];int size;
    }*root[N],*null,mem[N<<5];int tot;
    inline void init()
    {
        root[0]=null=new node();
        null->ch[0]=null->ch[1]=null;
    }
    inline void ins(node *&o,node *old,int l,int r,int pos)
    {
        o=mem+(tot++);o->size=old->size+1;
        if(l==r)return;RG int mi=l+r>>1;
        if(pos<=mi)o->ch[1]=old->ch[1],ins(o->ch[0],old->ch[0],l,mi,pos);
        else o->ch[0]=old->ch[0],ins(o->ch[1],old->ch[1],mi+1,r,pos);
    }
    inline void ins(int x,int y){ins(root[x],root[y],1,n,col[x]);}
    inline int querysz(node *a,node *b,int l,int r,int x)
    {
        if(l==r)return b->size - a->size;
        RG int mi=l+r>>1;
        if(x<=mi)return querysz(a->ch[0],b->ch[0],l,mi,x);
        return querysz(a->ch[1],b->ch[1],mi+1,r,x);
    }
    inline bool nope(int l,int r,int x)
        {return querysz(root[fa[l]],root[r],1,n,x)==0;}
}
namespace sgt1
{
    struct node
    {
        node *ch[2];int size;
        LL sum1,sum2,sum3;
    }*root[N],*null,mem[N<<5];int tot;
    inline LL S(int r){return r*(r+1ll)/2;}
    inline LL S2(int x){return (LL)x*(x+1)*(2*x+1)/6;}
    inline void ins(node *&o,node *old,int l,int r,int pos,int pr)
    {
        o=mem+(tot++);
        o->size=old->size+1;o->sum1=old->sum1+pos;
        o->sum2=old->sum2+pr;o->sum3=old->sum3+(LL)pos*pr;
        if(l==r)return;
        RG int mi=l+r>>1;
        if(pr<=mi)o->ch[1]=old->ch[1],ins(o->ch[0],old->ch[0],l,mi,pos,pr);
        else o->ch[0]=old->ch[0],ins(o->ch[1],old->ch[1],mi+1,r,pos,pr);
    }
    inline int querysz(node *a,node *b,int l,int r,int R)
    {
        if(l==r)return b->size - a->size;
        RG int mi=l+r>>1;
        if(mi<R)return b->ch[0]->size - a->ch[0]->size + querysz(a->ch[1],b->ch[1],mi+1,r,R);
        return querysz(a->ch[0],b->ch[0],l,mi,R);
    }
    inline void query(node *a,node *b,int l,int r,int R)
    {
        if(l==r)
        {
            sz+=b->size - a->size;s1+=b->sum1 - a->sum1;
            s2+=b->sum2 - a->sum2;s3+=b->sum3 - a->sum3;
            return;
        }
        RG int mi=l+r>>1;
        if(mi<R)
        {    
            sz+=b->ch[0]->size - a->ch[0]->size;s1+=b->ch[0]->sum1 - a->ch[0]->sum1;
            s2+=b->ch[0]->sum2 - a->ch[0]->sum2;s3+=b->ch[0]->sum3 - a->ch[0]->sum3;
            query(a->ch[1],b->ch[1],mi+1,r,R);
        }
        else query(a->ch[0],b->ch[0],l,mi,R);
    }
    inline int deal1(int l,int r)
        {return querysz(root[fa[l]],root[r],0,n-1,deep[fa[l]]);}
    inline LL q1(int A,int B)
    {
        if(A==B)return 0;
        sz=s1=s2=s3=0;query(root[A],root[B],0,n-1,A);
        A=deep[A],B=deep[B];
        return (LL)sz*A*B + (sz-s1)*A -s2*(B+1) + s3;//s1还是应该维护，这个地方没法算
    }
    inline LL exq1(int A,int B,int l)
    {
        if(A==B)return 0;
        sz=s1=s2=s3=0;query(root[fa[A]],root[B],0,n-1,fa[A]);
        //这里应该是要去fa[A]那里，lca那个地方也包含在区间里面，并且lca不能作为B的落点被算进去……？（lca之前在B侧的1--lca被算了）
        return ( sz*(deep[B]+1ll) - s1 )*l -l;
    }
    inline LL q2(int A,int B)
        {return ( root[A]->sum3 - S2(deep[A]) )*2 + (deep[A]+deep[B]+2ll)*( S(deep[A])-root[A]->sum2 ) - deep[A];}
    inline LL exq2(int A,int l)
        {return ( S(deep[A])-root[A]->sum2 ) * l;}
    inline LL deal2(int l,int r)
        {return ( (l<r)?q1(l,r):0 )+q2(l,r);}
    inline void work()
        {for(RG int i=1;i<=n;++i)ins(root[i],root[i-1],0,n-1,i,pre[i]);}
    inline void init()
        {root[0]=null=new node();null->ch[0]=null->ch[1]=null;}
    inline void ins(int x,int y)
        {ins(root[x],root[y],0,n-1,deep[x],pre[x]);}
}
int dfl[N],dfr[N],num;
inline void dfs(int rt)
{
    dfl[rt]=++num;
    RG int i,cur=lst[col[rt]];
    pre[rt]=lst[col[rt]];lst[col[rt]]=deep[rt];
    sgt1::ins(rt,fa[rt]);sgt2::ins(rt,fa[rt]);
    for(i=adj[rt];i;i=s[i].next)dfs(s[i].zhong);
    lst[col[rt]]=cur;
    dfr[rt]=num;
}
int vis[N],T;
inline int deal1(int l,int r)
{
    ++T;
    RG int fa1=l,fa2=r,ret=0;
    while(fa1>m)fa1=fa[fa1];
    while(fa2>m)fa2=fa[fa2];
    if(fa1==fa2)
    {
        while(l^r)
            if(deep[l] > deep[r])
                {if(vis[col[l]]!=T)vis[col[l]]=T,++ret;l=fa[l];}
            else
                {if(vis[col[r]]!=T)vis[col[r]]=T,++ret;r=fa[r];}
        if(vis[col[l]]!=T)vis[col[l]]=T,++ret;
        return ret;
    }
    if(fa1 > fa2)l^=r,r^=l,l^=r,fa1^=fa2,fa2^=fa1,fa1^=fa2;
    ret=sgt1::deal1(fa1,r);
    while(l>fa1)
    {
        if( vis[col[l]]!=T && sgt2::nope(fa1,r,col[l]) )
            vis[col[l]]=T,++ret;
        l=fa[l];
    }
    return ret;
}
int sta[30],cnt;
#include <vector>
vector<int>cont[N];
inline int calc(int pos,int lca,int r,int x)
{
    if(col[lca]==x)return 0;
    int ret=deep[r];
    for(vector<int>::iterator it=cont[x].begin();it!=cont[x].end();++it)
        if(deep[*it] >=deep[lca] && dfl[*it]<=dfl[r] && dfr[r] <=dfr[*it])
            ret=min(ret,deep[*it]-1);
    return ret-deep[lca];
}
inline LL deal2(int l,int r)
{
    ++T;
    RG int fa1=l,fa2=r,lca;
    LL ret=0;
    while(fa1>m)fa1=fa[fa1];
    while(fa2>m)fa2=fa[fa2];
    if(fa1==fa2)
    {
        for(fa1=l,fa2=r;l^r;)
            if(deep[l] > deep[r])l=fa[l];else r=fa[r];
        lca=l;l=fa1,r=fa2;
    }
    else 
    {
        if(fa1 > fa2)l^=r,r^=l,l^=r,fa1^=fa2,fa2^=fa1,fa1^=fa2;
        lca=fa1;
    }
    int l1=deep[l]-deep[lca];
    ret=sgt1::deal2(lca,r) //A在1---r all
    + sgt1::q1(lca,l) //A在lca---l，B在1--lca A侧的贡献
    + sgt1::exq2(lca,l1) //A在lca---l，B在1--lca B侧的贡献
    + sgt1::exq1(lca,r,l1);//A在lca---l，B在lca--r B侧的贡献
    cnt=0;
    fa1=l;
    while(l>lca)sta[++cnt]=l,l=fa[l];
    for(RG int i=cnt;i;--i)
    {
        l=sta[i];
        if(vis[col[l]]!=T)
            vis[col[l]]=T,ret+=(LL)calc(l,lca,r,col[l])*(deep[fa1]-deep[l]+1);
    }
    return ret;
}
int main()
{
    RG int i,j,t,l,r,opt;
    scanf("%d",&t);
    sgt1::init();
    sgt2::init();
    while(t--)
    {
        scanf("%d%d%u%u%u",&n,&m,&SA,&SB,&SC);
        for(i=2;i<=m;++i)fa[i]=i-1;
        for(i=m+1;i<=n;++i)fa[i]=make()%(i-1)+1;
        for(i=1;i<=n;++i)col[i]=make()%n+1;
        for(i=1;i<=n;++i)cont[i].clear();
        for(i=1;i<=n;++i)cont[col[i]].push_back(i);
        memset(lst,0,n+1<<2);
        num=0;
        if(m==n)
        {
            for(i=1;i<=n;++i)
                pre[i]=lst[col[i]],lst[col[i]]=i;
            sgt1::tot=0;
            sgt1::work();
            scanf("%d",&q);
            for(i=1;i<=n;++i)deep[i]=i;
            for(i=1;i<=q;++i)
            {
                scanf("%d%d%d",&opt,&l,&r);
                if(l>r)l^=r,r^=l,l^=r;
                if(opt==1)printf("%d
",sgt1::deal1(l,r));
                else printf("%lld
",sgt1::deal2(l,r));
            }
        }
        else
        {
            e=0;memset(adj,0,sizeof(adj));
            for(i=1;i<=n;++i)deep[i]=deep[fa[i]]+1;
            for(i=2;i<=n;++i)add(fa[i],i);
            sgt1::tot=sgt2::tot=0;dfs(1);
            scanf("%d",&q);
            for(i=1;i<=q;++i)
            {
                scanf("%d%d%d",&opt,&l,&r);
                if(l>r)l^=r,r^=l,l^=r;
                if(opt==1)printf("%d
",deal1(l,r));
                else printf("%lld
",deal2(l,r));
            }
        }
    }
}