LOJ #6284. 数列分块入门 8-分块(区间查询等于一个数c的元素，并将这个区间的所有元素改为c)

#6284. 数列分块入门 8

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

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

上传者： hzwer

提交提交记录统计测试数据讨论

2

 

题目描述

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

输入格式

第一行输入一个数字 nn。

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

接下来输入 nn 行询问，每行输入三个数字 ll、rr、cc，以空格隔开。

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

输出格式

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

样例

样例输入

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

样例输出

1
1
0
2

数据范围与提示

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

代码:

 //#6284. 数列分块入门 8-区间查询等于一个数c的元素，并将这个区间的所有元素改为c
 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const int maxn=1e6+;

 int n,m;
 int a[maxn],pos[maxn],tag[maxn];

 int Find(int l,int r,int c)
 {
     int num=;
     if(pos[l]==pos[r]){
         if(tag[pos[l]]==c) num+=r-l+;
         else{
             if(tag[pos[l]]!=-){
                 for(int i=(pos[l]-)*m+;i<=pos[l]*m;i++)
                     a[i]=tag[pos[l]];
             }
             for(int i=l;i<=r;i++){
                 if(a[i]==c) num++;
                 else a[i]=c;
             }
             int flag=;
             for(int i=(pos[l]-)*m+;i<=pos[l]*m;i++){
                 if(a[i]!=c){
                     flag=;break;
                 }
             }
             if(!flag) tag[pos[l]]=c;
             else tag[pos[l]]=-;
         }
     }
     else{
         if(tag[pos[l]]==c){
             num+=pos[l]*m-l+;
         }
         else{
             if(tag[pos[l]]!=-){
                 for(int i=(pos[l]-)*m+;i<=pos[l]*m;i++)
                     a[i]=tag[pos[l]];
             }
             for(int i=l;i<=pos[l]*m;i++){
                 if(a[i]==c) num++;
                 else a[i]=c;
             }
             int flag=;
             for(int i=(pos[l]-)*m+;i<=pos[l]*m;i++){
                 if(a[i]!=c){
                     flag=;break;
                 }
             }
             if(!flag) tag[pos[l]]=c;
             else tag[pos[l]]=-;
         }
         for(int i=pos[l]+;i<pos[r];i++){
             if(tag[i]==c) num+=m;
             else{
                 if(tag[i]!=-) tag[i]=c;
                 else{
                     for(int j=(i-)*m+;j<=i*m;j++){
                         if(a[j]==c) num++;
                         else a[j]=c;
                     }
                     int flag=;
                     for(int j=(i-)*m+;j<=i*m;j++){
                         if(a[j]!=c){
                             flag=;break;
                         }
                     }
                     if(!flag) tag[i]=c;
                     else tag[i]=-;
                 }
             }
         }
         if(tag[pos[r]]==c) num+=r-((pos[r]-)*m+)+;
         else{
             if(tag[pos[r]]!=-){
                 for(int i=(pos[r]-)*m+;i<=min(pos[r]*m,n);i++){
                     a[i]=tag[pos[r]];
                 }
             }
             for(int i=(pos[r]-)*m+;i<=r;i++){
                 if(a[i]==c) num++;
                 else a[i]=c;
             }
             int flag=;
             for(int i=(pos[r]-)*m+;i<=min(pos[r]*m,n);i++){
                 if(a[i]!=c){
                     flag=;break;
                 }
             }
             if(!flag) tag[pos[r]]=c;
             else tag[pos[r]]=-;
         }
     }
     return num;
 }

 int main()
 {
     scanf("%d",&n);
     m=sqrt(n);
     for(int i=;i<=n;i++){
         scanf("%d",&a[i]);
         pos[i]=(i-)/m+;
     }
     memset(tag,-,sizeof(tag));
     for(int i=;i<=n;i++){
         int l,r,c;
         scanf("%d%d%d",&l,&r,&c);
         printf("%d\n",Find(l,r,c));
     }
 }

 /*
 10
 1 3 4 2 5 7 11 3 5 1
 1 2 1
 2 4 2
 3 5 3
 4 6 4
 5 9 4
 3 5 1
 7 9 4
 5 9 4
 3 7 2
 5 6 2

 1
 1
 0
 0
 2
 0
 3
 4
 0
 2

 */