BZOJ 1878

求区间有多少不同的数

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

先写一个莫队的做法

代码如下:

#include
#include
#include
#include
#include
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.rq[i].l) {add(a[--l]);}
        while(rq[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
#include
#include
#include
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]));
    }
}

 

你可能感兴趣的:(莫队,主席树)