loj#6284. 数列分块入门 8

#6284. 数列分块入门 8

题目：传送门 

简要题意：

　　 给出一个长为 n 的数列，以及 n 个操作，操作涉及区间询问等于一个数 c 的元素，并将这个区间的所有元素改为 c。

---

题解：

　　 超级超级超级恶心！！！

　　 码农就算了...lazy的运用搞得我几乎要吐：

　　 lazy就看看该块是不是同一数字嘛...不是的话就为-1，是的话就更新成该块的数字

　　 然后又到暴力表演了，不停分情况（其实也就三种左右），头尾+中间　　 

---

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#define mod 10007
#define qread(x) x=read()
using namespace std;
typedef long long LL;
inline LL read()
{
    LL x=0,f=1;char ch;
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return f*x;
}
LL n,a[110000],ba[110000];
LL block,pos[110000];
void sol(LL l,LL r,LL c)
{
    LL ans=0;
    if(pos[l]==pos[r])
    {
        if(ba[pos[l]]==c)printf("%lld
",r-l+1);
        else
        {
            if(ba[pos[l]]!=-1)
                for(int i=(pos[l]-1)*block+1;i<=pos[l]*block;i++)
                    a[i]=ba[pos[l]];
            for(int i=l;i<=r;i++){if(a[i]==c)ans++;a[i]=c;}
            bool bk=false;
            for(int i=(pos[l]-1)*block+1;i<=pos[l]*block;i++)
                if(a[i]!=c){bk=true;break;}
            if(bk==false)ba[pos[l]]=c;
            else ba[pos[l]]=-1;
            printf("%lld
",ans);
        }
    }
    else
    {
        for(int i=pos[l]+1;i<=pos[r]-1;i++)
        {
            if(ba[i]==c)ans+=(i*block-((i-1)*block+1)+1);
            else if(ba[i]!=-1)ba[i]=c;
            else
            {
                for(int j=(i-1)*block+1;j<=i*block;j++)
                {
                    if(a[j]==c)ans++;
                    a[j]=c;
                }
                ba[i]=c;
            }
            continue;
        }
        if(ba[pos[l]]==c)ans+=(pos[l]*block-l+1);
        else 
        {
            if(ba[pos[l]]!=-1)
                for(int i=(pos[l]-1)*block+1;i<=pos[l]*block;i++)
                    a[i]=ba[pos[l]];
            for(int i=l;i<=pos[l]*block;i++)
            {
                if(a[i]==c)ans++;
                a[i]=c;
            }
            bool bk=false;
            for(int i=(pos[l]-1)*block+1;i<=pos[l]*block;i++)
                if(a[i]!=c){bk=true;break;}
            if(bk==false)ba[pos[l]]=c;
            else ba[pos[l]]=-1;
        }
        if(ba[pos[r]]==c)ans+=(r-((pos[r]-1)*block+1)+1);
        else 
        {
            if(ba[pos[r]]!=-1)
                for(int i=(pos[r]-1)*block+1;i<=pos[r]*block;i++)
                    a[i]=ba[pos[r]];
            for(int i=(pos[r]-1)*block+1;i<=r;i++)
            {
                if(a[i]==c)ans++;
                a[i]=c;
            }
            bool bk=false;
            for(int i=(pos[r]-1)*block+1;i<=pos[r]*block;i++)
                if(a[i]!=c){bk=true;break;}
            if(bk==false)ba[pos[r]]=c;
            else ba[pos[r]]=-1;
        }
        printf("%lld
",ans);
    }
}
int main()
{
    qread(n);for(int i=1;i<=n;i++)qread(a[i]);
    block=sqrt(n);
    for(int i=1;i<=n;i++)
    {    
        pos[i]=(i-1)/block+1;
    }
    memset(ba,-1,sizeof(ba));
    for(int i=1;i<=n;i++)
    {
        LL l,r,c;qread(l);qread(r);qread(c);
        sol(l,r,c);
    }
    return 0;
}