洛谷模板题:P1903
数据最多的那几组测试点超时了,但是懒得优化,正赛是不会卡常的(确信)。
记一笔:排序后、处理所有查询前那个 left=1,righ=0 的初始化很特别,因为一开始[left,right]是应该不包含任何区间,也就不包含任何颜色的。但是处理第一个询问时,left、right的移动可能会导致多余的颜色被计数,这时令left=1,[1, left-1]的颜色会被"减去",但是right右移的时候[0+1,right]之间的颜色会被“加上”,这样[1, left-1]之间的就被抵消了。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
struct Query
{
int left,right,no,t;
Query(int a=0,int b=0,int c=0,int d=0){ left=a; right=b; t=c; no=d; }
};
struct Modify
{
int pos,val;
Modify(int a=0,int b=0){ pos=a; val=b; }
};
Query Q[2000010];
Modify C[2000010];
int block[2000010],a[2000010],Qnum,Cnum,ans[2000010],res;
int cnt[2000010];
bool comp(const Query & a,const Query & b)
{
if( block[a.left] != block[b.left] )return block[a.left] < block[b.left];
else if( a.right != b.right )return a.right < b.right;
else return a.t < b.t;
}
void Add(int val)
{
if( cnt[val] == 0 )res++;
cnt[val]++;
}
void Delete(int val)
{
cnt[val]--;
if(cnt[val] == 0)res--;
}
int main()
{
int n,m;
scanf("%d %d",&n,&m);
int util = pow(n,0.666667);
for(int i=1;i<=n;i++){
scanf("%d",a+i);
block[i] = (i-1)/util + 1;
}
char op[5];
int x,y;
Qnum = Cnum = 0;
for(int i=0;i<m;i++)
{
scanf("%s %d %d",op,&x,&y);
if(op[0]=='Q'){
Qnum++;
Q[Qnum] = Query(x,y,Cnum,Qnum);
}
else{
Cnum++;
C[Cnum] = Modify(x,y);
}
}
sort( Q+1, Q+Qnum+1, comp );
int left=1, right=0, cur=0;
res=0;
for(int i=1;i<=Qnum;i++)
{
while( left < Q[i].left ){ Delete(a[left]); left++; }
while( left > Q[i].left ){ --left; Add(a[left]); }
while( right < Q[i].right ){ ++right; Add(a[right]); }
while( right > Q[i].right ){ Delete(a[right]); right--; }
while( cur < Q[i].t ){
cur++;
if( Q[i].left <= C[cur].pos && C[cur].pos <= Q[i].right ){
Add( C[cur].val );
Delete( a[C[cur].pos] );
}
swap( a[C[cur].pos], C[cur].val );
}
while( cur > Q[i].t ){
if( Q[i].left <= C[cur].pos && C[cur].pos <= Q[i].right ){
Add( C[cur].val );
Delete( a[C[cur].pos] );
}
swap( a[C[cur].pos], C[cur].val );
cur--;
}
ans[ Q[i].no ] = res;
}
for(int i=1;i<=Qnum;i++)printf("%d\n",ans[i]);
}