1878: [SDOI2009]HH的项链

题目链接

题目大意：询问[l,r]中不同元素个数

题解：树状数组（比较神），或者莫队

我的收获：套路&&莫队第一题达成

树状数组

以每个数字第一次在区间中出现的点代表所有的点。
记录每个位置i的数字的前一个相同数字出现的位置pre[i]，没有前一个相同的pre[i]为0

把询问转化一下，[l,r]中出现一次等价于pre[]在[1,l-1]中
添加一个元素i后，将pre[i]+1,i-1，然后求个a[i].l的前缀和

每次添加元素的时候，将它上一次出现的位置+1的地方的值加1，然后在其本身位置+1的值打上个-1标记，这样就成了线性求值，也就是每次求getsum(a[i].l)

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define M 1000005

int n,m,C;
int a[M],ans[M],c[M],pre[M],last[M];

struct node{int l,r,id;}q[M];

bool cmp(node x,node y){return x.r<y.r;}

void updata(int x,int v){
    for(int i=x;i<=n;i+=i&(-i))
        c[i]+=v;
}

int query(int x){
    int ret=0;
    for(int i=x;i>0;i-=i&(-i))
        ret+=c[i];
    return ret;
}

void work()
{
    int now=0;
    for(int i=1;i<=m;i++)
    {
        while(now<q[i].r) now++,updata(pre[now]+1,1),updata(now+1,-1);//树状数组的第一个元素只能从1开始
        ans[q[i].id]=query(q[i].l);
    }
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}

void init()
{
    cin>>n;
    for(int i=1,x;i<=n;i++)
        scanf("%d",&a[i]),pre[i]=last[a[i]],last[a[i]]=i;
    cin>>m;
    for(int i=1;i<=m;i++)
        scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
    sort(q+1,q+m+1,cmp);
}

int main()
{
    init();
    work();
    return 0;
}

莫队

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define M 50500

int tp,now;
int n,m,blo;
int a[M],pos[M],cnt[M],ans[M*4],map[1000005];

struct node{int l,r,id;}q[M*4];

inline int ID(int x){return map[x]?map[x]:map[x]=++tp;}

bool cmp(node x,node y){return pos[x.l]<pos[y.l]||(pos[x.l]==pos[y.l]&&x.r<y.r);}

void update(int x,int v)
{
    cnt[x]+=v;
    if(v==1&&cnt[x]==1) now++;
    if(v==-1&&!cnt[x]) now--;
}

void work()
{
    int l=1,r=0;
    for(int i=1;i<=m;i++)
    {
        while(r<q[i].r) update(a[++r],1);
        while(l>q[i].l) update(a[--l],1);
        while(r>q[i].r) update(a[r--],-1);
        while(l<q[i].l) update(a[l++],-1);
        ans[q[i].id]=now;
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}

void init()
{
    cin>>n;blo=sqrt(n);
    for(int i=1,x;i<=n;i++){
        scanf("%d",&x);
        a[i]=ID(x);
        pos[i]=(i-1)/blo+1;
    }
    cin>>m;
    for(int i=1;i<=m;i++)
        scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
    sort(q+1,q+m+1,cmp);
}

int main()
{
    init();
    work();
    return 0;
}