【题目描述】
给出一个长为 n 的数列,以及 n 个操作,操作涉及区间询问等于一个数 c 的元素,并将这个区间的所有元素改为 c。
【输入格式】
第一行输入一个数字 n。
第二行输入 n 个数字,第 i 个数字为 ai,以空格隔开。
接下来输入 n 行询问,每行输入三个数字 l、r、c,以空格隔开。
表示先查询位于 [l,r] 的数字有多少个是 c,再把位于 [l,r] 的数字都改为 c。
【输出格式】
对于每次询问,输出一行一个数字表示答案。
【样例】
样例输入
4
1 2 2 4
1 3 1
1 4 4
1 2 2
1 4 2样例输出
1
1
0
2【数据范围与提示】
对于 100% 的数据,1<=n<=100000,-2^31<=other,ans<=2^31-1。
【源代码】
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<bitset>
#define EPS 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
const int MOD = 10000+7;
const int N = 1000000+5;
const int dx[] = {-1,1,0,0,-1,-1,1,1};
const int dy[] = {0,0,-1,1,-1,1,-1,1};
using namespace std;
int n;
int block,sum;//block为块的长度,sum为块的个数
int a[N];//存放数列元素
int pos[N];//pos记录第i个元素在第几个块中
int flag[N];//标记整块是否值相同
void init(int n){
block=sqrt(n);//块的长度
for(int i=1;i<=n;i++)//第i个元素在第几块中
pos[i]=(i-1)/block+1;
memset(flag,-1,sizeof(flag));//标记初始化
}
void reset(int x){//更新不完整块的值
if(flag[pos[x]]==-1)
return;
else{
for(int i=(pos[x]-1)*block+1;i<=min(pos[x]*block,n);i++)//更新元素
a[i]=flag[pos[x]];
}
flag[pos[x]]=-1;//更新标记
}
int query(int L,int R,int x){
int res=0;
reset(L);//更新左端不完整块
for(int i=L;i<=min(pos[L]*block,R);i++){//左边的边角料
if(a[i]==x)
res++;
a[i]=x;
}
if(pos[L]!=pos[R]){//存在右区间才遍历,防止重复计算
reset(R);//更新右端不完整块
for(int i=(pos[R]-1)*block+1;i<=R;i++){//右边的边角料
if(a[i]==x)
res++;
a[i]=x;
}
}
for(int i=pos[L]+1;i<=pos[R]-1;i++){//中间的整块
if(flag[i]==-1){//没有标记,暴力查询
for(int j=(i-1)*block+1;j<=i*block;j++){
if(a[j]==x)
res++;
a[j]=x;
}
}
else if(flag[i]==x)//整块相等
res+=block;
flag[i]=x;//更新标记
}
return res;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
init(n);
for(int i=1;i<=n;i++){
int left,right,x;
scanf("%d%d%d",&left,&right,&x);
printf("%d\n",query(left,right,x));
}
return 0;
}