BZOJ1878 HH的项链

BZOJ1878 HH的项链

题目大意

给定一个长度为n序列，共有q次询问，对于每次询问Li，Ri，输出区间[Li，Ri]内有
几个不重复的数。

数据范围

对于20%的数据，N ≤ 100，M ≤ 1000；
对于40%的数据，N ≤ 3000，M ≤ 200000；
对于100%的数据，N ≤ 50000，M ≤ 200000。

解决方法

• 离线做法
  1.把每次询问按右端点从小到大排序
  2.每次从上一个询问的右端点扫描到当前询问的右端点，对于扫描过程遇到的数，
  把当前位置的值加一，并把这个数之前最后一次出现的位置的值减一，（用数
  状数组维护）然后求出区间和就是答案了。
  3.总体思想就是保持每个数只保留最靠近区间右端点的一个，利用树状数组进行
  修改及求和（也可以用线段树）。

代码

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<iostream>
#include<queue>
#define N 50007
#define M 200007
#define lowbit(x) x&(-x)

int read(){
    int num=0;
    char c;
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9'){
        num=num*10+(c-'0');
        c=getchar();
    }
    return num;
}//读入优化

struct info{int l,r,ans;}que[M];//区间
bool cmp1(int a,int b){return que[a].r<que[b].r;}

int n,m,k,pos[20*N],last[N],c[N],id[M];

void adds(int x,int s){
  if (x==0) return;
  for (int i=x;i<=n;i+=lowbit(i)) 
   c[i]+=s;
}

int sigma(int x){
   int ans=0;
   for (int i=x;i>0;i-=lowbit(i))
    ans+=c[i];
   return ans;
}

int main(){
  freopen("diff.in","r",stdin);
  freopen("diff.out","w",stdout);
  n=read();int num;
  for (int i=1;i<=n;++i){
    num=read();                 //pos指向每个数最后一次出现的位置
    last[i]=pos[num];pos[num]=i;//last指向当前这个数上一次出现的位置
    adds(i,1);//我在这里便把所有位置的值加1，所以后面没有加1的操作
  }
  m=read();
  for (int i=1;i<=m;++i) id[i]=i,que[i].l=read(),que[i].r=read();
  std::sort(id+1,id+m+1,cmp1);k=1;//把区间按右端点从大到小排序
  for (int i=1;i<=m;++i){      
    for (int j=k;j<=que[id[i]].r;++j) adds(last[j],-1);//上一个位置减1
    k=que[id[i]].r+1;int suma,sumb;
    suma=sigma(que[id[i]].l-1);sumb=sigma(que[id[i]].r);//求前缀和
    que[id[i]].ans=sumb-suma;//通过前缀和相减得出区间和
  }
  for (int i=1;i<=m;++i)
   printf("%d\n",que[i].ans);
}

后记

• 在线做法
  1.其实在线做法与离线做法的思想是一样的，只不过在线做法是建了n棵线段树（即主席树），第i棵树存的是从1扫描到i的结果（线段树维护区间和）

• 下面给出在线做法的代码

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<iostream>
#include<queue>
#define N 50007
#define M 3000000//M随便开的，实际上不用这么大

int read(){
    int num=0;
    char c;
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9'){
        num=num*10+(c-'0');
        c=getchar();
    }
    return num;
}

int n,m,size,pos[20*N],last[N];
int root[N],s[M],ls[M],rs[M];

void insert(int l,int r,int x,int &y,int c,int add){
    y=++size;
    s[y]=s[x]+add;
    if (l==r) return;
    int mid=(l+r)>>1;
    ls[y]=ls[x];rs[y]=rs[x];
    if (c<=mid) insert(l,mid,ls[x],ls[y],c,add);
    else insert(mid+1,r,rs[x],rs[y],c,add);
}

int query(int l,int r,int x,int ql,int qr){
    if (l>=ql&&r<=qr) 
     return s[x];
    int mid=(l+r)>>1,ans=0;
    if (mid>=ql) ans+=query(l,mid,ls[x],ql,qr);
    if (qr>=mid+1) ans+=query(mid+1,r,rs[x],ql,qr);
    return ans;
}

int main(){
  freopen("diff.in","r",stdin);
  freopen("diff.out","w",stdout);
  n=read();int num;
  for (int i=1;i<=n;++i) num=read(),last[i]=pos[num],pos[num]=i;
  for (int i=1;i<=n;++i){
    insert(1,n,root[i-1],root[i],i,1);//当前位置加1
    if (last[i]!=0) insert(1,n,root[i],root[i],last[i],-1);
    //之前的位置减1
  }
  int l,r;
  for (m=read();m>0;--m){
    l=read();r=read();
    printf("%d\n",query(1,n,root[r],l,r));
  }
}

• 提交地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1878