LOJ#6284. 数列分块入门 8

#6284. 数列分块入门 8

内存限制：256 MiB时间限制：500 ms标准输入输出

题目类型：传统评测方式：文本比较

上传者： hzwer

提交提交记录统计讨论

1

测试数据

 

题目描述

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

输入格式

第一行输入一个数字 nnn。

第二行输入 nnn 个数字，第 i 个数字为 aia_ia​i​​，以空格隔开。

接下来输入 nnn 行询问，每行输入三个数字 lll、rrr、ccc，以空格隔开。

表示先查询位于 [l,r][l,r][l,r] 的数字有多少个是 ccc，再把位于 [l,r][l,r][l,r] 的数字都改为 ccc。

输出格式

对于每次询问，输出一行一个数字表示答案。

样例

样例输入

4
1 2 2 4
1 3 1
1 4 4
1 2 2
1 4 2

样例输出

1
1
0
2

数据范围与提示

对于 100% 100\%100% 的数据，1≤n≤100000,−231≤others 1 \leq n \leq 100000, -2^{31} \leq \mathrm{others}1≤n≤100000,−2​31​​≤others、ans≤231−1 \mathrm{ans} \leq 2^{31}-1ans≤2​31​​−1。

分类标签

hzwer 题集 分块及按大小分类 数据结构

 

 

维护每个块被修改成了什么值。

零散块下放标记后暴力

时间复杂度证明：

 

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAXN=1e5+10,mod=10007;
inline char nc()
{
    static char buf[MAXN],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,MAXN,stdin)),p1==p2?EOF:*p1++;
}
inline int read()
{
    char c=nc();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=nc();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=nc();}
    return x*f;
}
int a[MAXN],belong[MAXN],L[MAXN],R[MAXN],block,tag[MAXN];
void reset(int x)
{
    if(tag[x]==-1) return ;
    for(int i=L[x*block];i<=R[x*block];i++)
        a[i]=tag[x];
    tag[x]=-1;
}
int Query(int l,int r,int val)
{
    int ans=0;
    reset(belong[l]);
    for(int i=l;i<=min(r,R[l]);i++)
    {
        if(a[i]==val) ans++;
        a[i]=val;
    }
    if(belong[l]!=belong[r])
    {
        reset(belong[r]);
        for(int i=L[r];i<=r;i++)
        {
            if(a[i]==val) ans++;
            a[i]=val;
        }
    }
    for(int i=belong[l]+1;i<=belong[r]-1;i++)
    {
        if(tag[i]==-1)
        {
            for(int j=L[i*block];j<=R[i*block];j++)
                if(a[j]==val)
                    ans++;
                else a[j]=val;
        }
        else if(tag[i]==val) ans+=block;
        tag[i]=val;
    }
    return ans;
}
int main()
{
    #ifdef WIN32
    freopen("a.in","r",stdin);
    #else
    #endif
    int N=read();block=sqrt(N);
    memset(tag,-1,sizeof(tag));
    for(int i=1;i<=N;i++) a[i]=read(),belong[i]=(i-1)/block+1,L[i]=(belong[i]-1)*block+1,R[i]=belong[i]*block;
    for(int i=1;i<=N;i++)
    {
        int l=read(),r=read(),c=read();
        printf("%d\n",Query(l,r,c));
    }
    return 0;
}