BZOJ 1878: [SDOI2009]HH的项链 树状数组+乱搞

时空隧道

分析：

以前看过一道类似的题目（具体是啥不记得了）….所以这次还有点思路…

我们考虑如果没有重复的数字直接算就好了…有重复的数字就应该每个数字只计算一次…怎么实现这种思想….

考虑对于一个询问，第一次出现的数字就是最靠左的数字，那么我们就把这个区间中第一次出现的数字设为1，求个前缀和就好了…

所以说我们可以离线做…把所有询问按照左端点排序…对于当前的询问答案就直接计算前缀和就好…计算完之后，对于s[i].l~s[i+1].l-1这些数字，insert(j,-1),insert(nxt[j],1)，这样就保证了对于当前的询问数组中设为1的数字都是不重复的….而且也不会漏掉一些数字…

代码如下：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//by NeighThorn
using namespace std;
const int maxn=50000+5,maxm=1000000+5;
int n,m,a[maxn],tr[maxn],pos[maxm],nxt[maxn];
struct M{
int l,r,id,ans;
}s[200000+5];
inline bool cmp1(M a,M b){
if(a.l==b.l)
return a.r<b.r;
return a.l<b.l;
}
inline bool cmp2(M a,M b){
return a.id<b.id;
}
inline void insert(int x,int y){
for(;x<=1000000;x+=x&(-x))
tr[x]+=y;
}
inline int query(int x){
int sum=0;
for(;x;x-=x&(-x))
sum+=tr[x];
return sum;
}
signed main(void){
scanf("%d",&n);memset(tr,0,sizeof(tr));
memset(pos,0,sizeof(pos));
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
if(pos[a[i]]==0)
insert(i,1),pos[a[i]]=i;
else
nxt[pos[a[i]]]=i,pos[a[i]]=i;
}
scanf("%d",&m);
for(int i=1;i<=m;i++)
scanf("%d%d",&s[i].l,&s[i].r),s[i].id=i;
sort(s+1,s+m+1,cmp1);s[0].l=0;
for(int i=1;i<=m;i++){
//  cout<<s[i].l<<" "<<s[i].r<<endl;
s[i].ans=query(s[i].r);
//  cout<<s[i].ans<<endl;
if(i<m&&s[i].l!=s[i+1].l)
for(int j=s[i].l;j<s[i+1].l;j++){
insert(j,-1);
if(nxt[j])
insert(nxt[j],1);
}
}
sort(s+1,s+m+1,cmp2);
for(int i=1;i<=m;i++)
printf("%d\n",s[i].ans);
return 0;
}

by >_< NeighThorn