BZOJ 1878

求区间有多少不同的数

可以莫队，离线树状数组，主席树做

先写一个莫队的做法

代码如下：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n,m,l,r,visit[1000005],pos[50005],a[50005],sz,ans,sum[200005];
struct Node
{
    int l,r,id;
}q[200005];
bool cmp(Node a,Node b)
{
    if(pos[a.l]==pos[b.l]) return a.r<b.r;
    else return a.l<b.l;
}
bool cmp_id(Node a,Node b)
{
    return a.id<b.id;
}
void add(int p)
{
    if(!visit[p]) ans++;
    visit[p]++;
}
void del(int p)
{
    visit[p]--;
    if(!visit[p]) ans--;
}
int main()
{
    scanf("%d",&n); sz=sqrt(n);
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
        pos[i]=(i-1)/sz+1;
    }
    scanf("%d",&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+1+m,cmp);
    int l=1,r=0;
    for(int i=1;i<=m;i++) {
        while(l<q[i].l) {del(a[l++]);}
        while(l>q[i].l) {add(a[--l]);}
        while(r<q[i].r) {add(a[++r]);}
        while(r>q[i].r) {del(a[r--]);}
        sum[q[i].id]=ans;
    }
    for(int i=1;i<=m;i++) {
        printf("%d\n",sum[i]);
    }
}

主席树

一直WA，一直没找到错在哪里，最后发现last开小了 QAQ 啊啊

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=60000;
int n,m,last[1000005],root[N],a[N],cnt,temp,x,y;
struct Node
{
    int l,r,sum;
}p[N*40];
void Update(int l,int r,int &x,int y,int pos,int v)
{
    p[++cnt]=p[y]; p[cnt].sum+=v; x=cnt;
    if(l==r) {
        return ;
    }
    int mid=l+r>>1;
    if(pos<=mid) {
        Update(l,mid,p[x].l,p[y].l,pos,v);
    }
    else {
        Update(mid+1,r,p[x].r,p[y].r,pos,v);
    }
}
int Query(int l,int r,int pos,int x)
{
//    printf("%d %d\n",pos,x);
    if(l==r) {
        return p[x].sum;
    }
    int mid=l+r>>1;
    if(pos<=mid) {
        return p[p[x].r].sum+Query(l,mid,pos,p[x].l);
    }
    else{
        return Query(mid+1,r,pos,p[x].r);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=n;i++) {
        int v=a[i];
        if(last[v]>0) {
            Update(1,n,temp,root[i-1],last[v],-1);
            Update(1,n,root[i],temp,i,1);
        }
        else{
            Update(1,n,root[i],root[i-1],i,1);
        }
        last[v]=i;
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++) {
        scanf("%d%d",&x,&y);
        printf("%d\n",Query(1,n,x,root[y]));
    }
}