数列分块入门 8

题目链接
给出一个长为n的数列，以及n个操作，操作涉及区间询问等于一个数c的元素，并将这个区间的所有元素改为c。
我们用一个标记记录这个块中的元素是不是都一样，一样的话进行标记，修改时在该块中的直接修改标记就好了，不在该块内的进行暴力更改。

#include<set>
#include<map>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<string>
#include<queue>
using namespace std;
typedef long long int ll;
const int N=100000+10;
int n,m;
int v[N],pos[N],tag[N];
void reset(int x)
{
    if(tag[x]==-1)return ;
    for(int i=(x-1)*m+1;i<=min(x*m,n);i++)
        v[i]=tag[x];
    tag[x]=-1;
}
int solve(int l,int r,int c)
{
    int ans=0;
    reset(pos[l]);
    for(int i=l;i<=min(pos[l]*m,r);i++)
        if(v[i]!=c)v[i]=c;
        else ans++;
    if(pos[l]!=pos[r])
    {
        reset(pos[r]);
        for(int i=(pos[r]-1)*m+1;i<=r;i++)
            if(v[i]!=c)v[i]=c;
            else ans++;
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
        if(tag[i]!=-1)
        {
            if(tag[i]!=c)tag[i]=c;
            else ans+=m;
        }
        else
        {
            for(int j=(i-1)*m+1;j<=i*m;j++)
                if(v[j]!=c)v[j]=c;
                else ans++;
            tag[i]=c;
        }
    return ans;
}
int main()
{
    scanf("%d",&n);
    m=sqrt(n);
    memset(tag,-1,sizeof(tag));
    for(int i=1;i<=n;i++)scanf("%d",v+i);
    for(int i=1;i<=n;i++)pos[i]=(i-1)/m+1;
    for(int i=1;i<=n;i++)
    {
        int l,r,c;
        scanf("%d%d%d",&l,&r,&c);
        printf("%d\n",solve(l,r,c));
    }
    return 0;
}