题目:
题解:
有一个非常重要的思路:我们给每一个数弄一个pre[i]表示前面那个跟自已一个种类的位置,那么在查询[l,r]的时候,pre[i] < l的就是第一次出现啊!
那么在查询的时候,分块处理,块外的暴力,块内的给pre排序二分处理;修改的时候,根据pre和nxt的关系暴力修改
要特别注意change的时候在什么情况下修改nxt/pre,什么时候不修改,也要注意insert的时候是pos[i]而不是i。。。
代码:
#include <algorithm>
#include <cstdio>
#include <cmath>
using namespace std;
const int N=10005;
int n,m,cnt,a[N],col[1000005],pre[N],nxt[N],b[N],pos[N];
void insert(int x)
{
int l=(x-1)*cnt+1,r=min(n,x*cnt);
for (int i=l;i<=r;i++) b[i]=pre[i];
sort(b+l,b+r+1);
}
int find(int x,int w)
{
int l=(x-1)*cnt+1,r=min(n,x*cnt),L=(x-1)*cnt+1,ans=L-1;
while (l<=r)
{
int mid=(l+r)>>1;
if (b[mid]<w) ans=mid,l=mid+1;else r=mid-1;
}
return ans-L+1;
}
int qurry(int l,int r)
{
int ans=0;
if (pos[l]==pos[r])
{
for (int i=l;i<=r;i++)
if (pre[i]<l) ans++;
}
else
{
int up=min(n,pos[l]*cnt);
for (int i=l;i<=up;i++)
if (pre[i]<l) ans++;
up=(pos[r]-1)*cnt+1;
for (int i=up;i<=r;i++)
if (pre[i]<l) ans++;
for (int i=pos[l]+1;i<pos[r];i++) ans+=find(i,l);
}
return ans;
}
void change(int x,int c)
{
int ch=nxt[x];
if (ch!=n+1) pre[nxt[x]]=pre[x],insert(pos[ch]);
if (pre[x]) nxt[pre[x]]=nxt[x];
int wh=n+1;
for (int i=x+1;i<=n;i++)
if (a[i]==c){wh=i;break;}
if (wh!=n+1) pre[wh]=x,insert(pos[wh]);
nxt[x]=wh;
wh=0;
for (int i=x-1;i>=1;i--)
if (a[i]==c){wh=i;break;}
if (wh) nxt[wh]=x;
pre[x]=wh,insert(pos[x]);
a[x]=c;
}
int main()
{
int q;scanf("%d%d",&n,&q);
cnt=sqrt(n);
for (int i=1;i<=n;i++) scanf("%d",&a[i]);
for (int i=1;i<=n;i++) pos[i]=(i-1)/cnt+1,col[a[i]]=0;
for (int i=1;i<=n;i++) pre[i]=col[a[i]],col[a[i]]=i;
for (int i=1;i<=n;i++) col[a[i]]=n+1;
for (int i=n;i>=1;i--) nxt[i]=col[a[i]],col[a[i]]=i;
if (n%cnt) m=n/cnt+1;else m=n/cnt;
for (int i=1;i<=m;i++) insert(i);
while (q--)
{
char st[5];scanf("%s",st);
int l,r;scanf("%d%d",&l,&r);
if (st[0]=='Q') printf("%d\n",qurry(l,r));
else change(l,r);
}
}