BZOJ 2120 树状数组套平衡树

题解：

每个数第一次出现时随便改成pos-n，之后改成前一次出现的位置。询问的时候查找一个区间中比左端点小的数的个数

吐槽：

尼玛数组越界了查了一晚上+半个下午。

树套树这种东西真难查错！

我多插入了很多没有用的节点，所以就慢了点，不过省代码~

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cstdio>
 
#define N 6000050
#define INF 520131499
#define CN 1000000
#define lowbit(x) x&-x
#define BUG system("pause");
#define CA puts("ca");
 
using namespace std;
 
int sz[N],val[N],fa[N],son[N][2];
int rc[CN+10],root[N],col[N];
int cnt,num,q[N+CN];
int suf[N],pre[N],jp[CN+10];
int n,m;

inline void prt(int x)
{
    if(!x) return;
    prt(son[x][0]);
    printf("%d    ",val[x]);
    prt(son[x][1]);
}

inline void pushup(int x)
{
    if(!x) return;
    sz[x]=sz[son[x][0]]+sz[son[x][1]]+1;
}
 
inline void link(int x,int y,int c)
{
    fa[x]=y; son[y][c]=x;
}
 
inline void rotate(int x,int c)
{
    int y=fa[x];
    link(x,fa[y],son[fa[y]][1]==y);
    link(son[x][!c],y,c);
    link(y,x,!c);
    pushup(y);
}
 
inline void splay(int x,int g,int &rt)
{
    while(fa[x]!=g)
    {
        int y=fa[x];
        int cy=(son[fa[y]][1]==y),cx=(son[y][1]==x);
        if(fa[y]==g) rotate(x,cx);
        else
        {
            if(cx==cy) rotate(y,cy);
            else rotate(x,cx);
            rotate(x,cy);
        }
    }
    pushup(x);
    if(!g) rt=x;
}
 
inline int getnum()
{
    if(num==0) return ++cnt;
    return q[num--];
}
 
inline void newnode(int y,int &x,int sp)
{
    x=getnum();
    fa[x]=y; val[x]=sp; sz[x]=1;
    son[x][1]=son[x][0]=0;
}
 
inline int getmin(int x)
{
    while(son[x][0]) x=son[x][0];
    return x;
}
 
inline int getmax(int x)
{
    while(son[x][1]) x=son[x][1];
    return x;
}
 
inline int pred(int x,int sp)
{
    int res=-INF-10,ans=x;
    while(x)
    {
        if(val[x]==sp) return x;
        else if(val[x]<sp&&val[x]>res) res=val[x],ans=x;
        x=son[x][val[x]<sp];
    }
    return ans;
}
 
inline int succ(int x,int sp)
{
    int res=INF+10,ans=x;
    while(x)
    {
        if(val[x]==sp) return x;
        else if(val[x]>sp&&val[x]<res) res=val[x],ans=x;
        x=son[x][val[x]<sp];
    }
    return ans;
}

inline int query(int &rt,int sp)
{
    int l=getmin(rt),r=succ(rt,sp);
    splay(l,0,rt); splay(r,l,rt);
    return sz[son[r][0]];
}
 
inline void insert(int &rt,int sp)
{
    int x=rt;
    while(son[x][val[x]<sp]) x=son[x][val[x]<sp];
    newnode(x,son[x][val[x]<sp],sp);
    splay(son[x][val[x]<sp],0,rt);
}
 
inline void del(int &rt,int sp)
{
    int p=pred(rt,sp-1),s=succ(rt,sp+1);
    splay(p,0,rt); splay(s,p,rt);
    
    q[++num]=son[s][0];
    son[s][0]=0;
    pushup(s); pushup(p);
}
 
inline int getans(int x,int sp)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=query(root[i],sp);
    return res;
}
 
inline void updata(int x,int y)
{
    if(y==col[x]) return;
    for(int i=x;i<=n;i+=lowbit(i)) del(root[i],pre[x]);//删除当前颜色的前缀 
    if(suf[x]<=n)
    {
        for(int i=suf[x];i<=n;i+=lowbit(i)) del(root[i],pre[suf[x]]);//删除当前颜色后缀的前缀 
        pre[suf[x]]=pre[x];//把当前颜色后缀的前缀改为当前颜色 
        if(pre[suf[x]]<=0) pre[suf[x]]=suf[x]-n;
        for(int i=suf[x];i<=n;i+=lowbit(i)) insert(root[i],pre[suf[x]]);//把当前颜色后缀的前缀改为当前颜色 
    }
    if(pre[x])
    {
        suf[pre[x]]=suf[x];//把当前颜色的前缀的后缀改为当前颜色的后缀 
        if(suf[pre[x]]>n) suf[pre[x]]=pre[x]+n;
    }
    del(rc[col[x]],x); insert(rc[y],x);//颜色splay中删除旧颜色，添加新颜色 
    int l=val[getmax(son[rc[y]][0])],r=val[getmin(son[rc[y]][1])];//新颜色的前缀和后缀（新） 
    if(l!=-INF) pre[x]=l,suf[l]=x;//前缀和当前建立关系 （新） 
    else pre[x]=x-n; 
    if(r!=INF)//存在后缀 （新） 
    {
        for(int i=r;i<=n;i+=lowbit(i)) del(root[i],pre[r]);//删除新颜色的后缀的前缀 （新） 
        suf[x]=r;pre[r]=x;//后缀和当前建立关系 
        for(int i=r;i<=n;i+=lowbit(i)) insert(root[i],pre[r]);//插入新颜色的前缀 （新） 
    }
    else suf[x]=n+x;
    for(int i=x;i<=n;i+=lowbit(i)) insert(root[i],pre[x]);//插入当前颜色的前缀 （新） 
    col[x]=y;
}
 
inline void read()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&col[i]);
        newnode(0,root[i],-INF);
        newnode(root[i],son[root[i]][1],INF);
        sz[root[i]]=2;
    }
    for(int i=1;i<=n;i++)
    {
        if(!jp[col[i]]) pre[i]=i-n;
        else pre[i]=jp[col[i]];
        jp[col[i]]=i;
    }
    for(int i=1;i<=CN;i++) jp[i]=n+1;
    for(int i=n;i>=1;i--)
    {
        if(jp[col[i]]==n+1) suf[i]=i+n;
        else suf[i]=jp[col[i]];
        jp[col[i]]=i;
    }
    for(int i=1;i<=CN;i++)
    {
        newnode(0,rc[i],-INF);
        newnode(rc[i],son[rc[i]][1],INF);
        sz[rc[i]]=2;
    }
}

inline void go()
{
    for(int i=1;i<=n;i++)
    {
        for(int j=i;j<=n;j+=lowbit(j)) insert(root[j],pre[i]);
        insert(rc[col[i]],i);
    }
    char str[3];int x,y;
    while(m--)
    {
        scanf("%s%d%d",str,&x,&y);
        if(str[0]=='Q') printf("%d\n",getans(y,x)-getans(x-1,x));
        else updata(x,y);
    }
}
 
int main()
{
    read(),go();
    return 0;
}

 赠送对拍器：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <ctime>

using namespace std;

int main()
{
    srand(time(0));
    while(1)
    {
        FILE *fp=fopen("3.txt","w");
        int n=10,m=20;
        fprintf(fp,"%d %d\n",n,m);
        for(int i=1;i<=n;i++) fprintf(fp,"%d ",rand()%6+1);
        fprintf(fp,"\n");
        for(int i=1;i<=m;i++)
        {
            int a,b,c;
            a=rand()&1;
            b=rand()%10+1;
            if(a)
            {
                c=rand()%10+1;
                if(b>c) swap(b,c);
                fprintf(fp,"Q %d %d\n",b,c);
            }
            else
            {
                c=rand()%6+1;
                fprintf(fp,"R %d %d\n",b,c);
            }
        }
        fclose(fp);
        system("1.exe");system("2.exe");
        if(system("fc 1.txt 2.txt"))
        {
            puts("ca");
            getchar();
        }
    }
    return 0;
}