[Ynoi2018] 五彩斑斓的世界

由乃OI的卡常题......

CF896E Welcome home, Chtholly 传送门

洛谷 CF896E 传送门

以上两个传送门是CF上的，3000ms、512Mb。

这个 洛谷 P4117 [Ynoi2018]五彩斑斓的世界 是1000ms、128Mb（卡常+卡空间）。

这道题的修改操作很特殊，是对区间内某些拥有特殊性质的数进行操作。

各种数据结构无法满足这个要求，所以我们考虑分块。

把块内值相等的数用并查集连到一起，每个值记录一个head就行了。

但是如果块内全是大于x的数，我们需要对所有的head进行操作，复杂度太高了。

所以如果有一半以上都是大于x的，考虑将不大于x的都加上x，再打个标记让整个块减少x就行了。

其实时间空间也不是太卡，稍微搞一搞就能过了。

#include<cstdio>
#include<cmath>
#include<algorithm>

using std::max;
using std::min;
typedef unsigned int ui;

ui n,m,sz;
ui v[100001],hv[100001];
ui f[100001],hd[225][100001];
ui l[225],r[225],cnt[100001];
ui bl[100001],mx[225],lz[225];

ui read()
{
    char c=getchar();ui ret=0u;
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')ret=ret*10u+c-'0',c=getchar();
    return ret;
}

void getmx(ui id)
{
    while(!hd[id][mx[id]])mx[id]--;
}

ui fin,bef;

ui fa(ui p)
{
    fin=p;
    while(fin^f[fin])fin=f[fin];
    while(f[p]^fin)bef=p,p=f[p],f[bef]=fin;
    return fin;
}

void reset(ui id)
{
    ui i;
    for(i=l[id];i<=r[id];i++)v[i]=hv[fa(i)],hd[id][v[i]]=0u;
    for(i=l[id];i<=r[id];i++)f[i]=i,cnt[i]=1u;
    for(i=l[id];i<=r[id];i++)
    {
        if(hd[id][v[i]])
            f[i]=hd[id][v[i]],cnt[hd[id][v[i]]]+=cnt[i];
        else
            hd[id][v[i]]=i,hv[i]=v[i];
    }
    getmx(id);
}

void change(ui id,ui lb,ui rb,ui x)
{
    ui i;
    for(i=l[id];i<=r[id];i++)v[i]=hv[fa(i)];
    for(i=l[id];i<=r[id];i++)hd[id][v[i]]=0u;
    for(i=lb;i<=rb;i++)if(v[i]-lz[id]>x)v[i]-=x;
    for(i=l[id];i<=r[id];i++)f[i]=i,hv[i]=v[i];
    reset(id);
}

int main()
{
    n=read(),m=read();sz=(ui)(sqrt(2*n));
    ui i,j;ui *fr,*to;
    for(i=1u;i<=n;i++){bl[i]=(i-1u)/sz+1u;f[i]=i;}
    for(i=1u;i<=bl[n];i++){l[i]=(i-1u)*sz+1u;r[i]=min(n,i*sz);}
    for(i=1u;i<=n;i++)
        {v[i]=hv[i]=read();mx[bl[i]]=max(mx[bl[i]],v[i]);}
    for(i=1u;i<=bl[n];i++)reset(i);
    ui op,lb,rb,x;
    while(m--)
    {
        op=read(),lb=read(),rb=read(),x=read();
        if(op==1u)
        {
            if(bl[lb]==bl[rb])
            {
                change(bl[lb],lb,rb,x);
                continue;
            }
            change(bl[lb],lb,r[bl[lb]],x);
            change(bl[rb],l[bl[rb]],rb,x);
            for(i=bl[lb]+1u;i<bl[rb];i++)
            {
                if((x<<1)>mx[i]-lz[i])
                {
                    for(j=x+lz[i]+1u;j<=mx[i];j++)
                    {
                        fr=&hd[i][j],to=&hd[i][j-x];
                        if(!(*fr))continue;
                        if(*to){cnt[*to]+=cnt[*fr];f[*fr]=*to;}
                        else{(*to)=(*fr);hv[*to]=j-x;}
                        *fr=0u;
                    }
                    getmx(i);
                }
                else
                {
                    for(j=lz[i]+1u;j<=x+lz[i];j++)
                    {
                        fr=&hd[i][j],to=&hd[i][j+x];
                        if(!(*fr))continue;
                        if(*to){cnt[*to]+=cnt[*fr];f[*fr]=*to;}
                        else{(*to)=(*fr);hv[*to]=j+x;}
                        *fr=0u;
                    }
                    lz[i]+=x;
                }
            }
        }
        if(op==2u)
        {
            ui ans=0u;
            if(bl[lb]==bl[rb])
            {
                for(i=lb;i<=rb;i++)
                    if(hv[fa(i)]==x+lz[bl[lb]])ans++;
                printf("%u
",ans);
                continue;
            }
            for(i=lb;i<=r[bl[lb]];i++)
                if(hv[fa(i)]==x+lz[bl[lb]])ans++;
            for(i=l[bl[rb]];i<=rb;i++)
                if(hv[fa(i)]==x+lz[bl[rb]])ans++;
            for(i=bl[lb]+1u;i<bl[rb];i++)
                if(x+lz[i]<=100000u)ans+=cnt[hd[i][x+lz[i]]];
            printf("%u
",ans);
        }
    }
    return 0;
}