求给定区间外有多少个不同的数(区间倍增+主席树)

思路:求区间不同个数自然很容易想到主席树/莫队,但是这里是区间外,怎么处理呢?求区间内不同的个数给定的是连续的范围,而区间外的则是一个断开的,所以应该要想到把两个区间拼接在一起。怎么拼接呢,令a[i+n]=a[i],即多保存一份副本在区间外,这样恰当,当求区间(1,L)和区间(R,,n)时实际上转化成求区间 (R,L+n),这样就把断开的区间变成连续的区间,用主席树或莫队解决。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+50;
int n,q,root[maxn],pre[maxn],a[maxn<<1],temp,cnt;

struct Tree{
    int ls,rs,sum;
}tree[maxn*40];

void update(int l,int r,int &x,int y,int pos,int add){
    tree[++cnt]=tree[y],x=cnt,tree[x].sum+=add;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(pos<=mid) update(l,mid,tree[x].ls,tree[y].ls,pos,add);
    else update(mid+1,r,tree[x].rs,tree[y].rs,pos,add);
}

int query(int l,int r,int x,int y){
    if(x<=l) return tree[y].sum;
    int mid=(l+r)>>1;
    if(x<=mid) return query(l,mid,x,tree[y].ls)+tree[tree[y].rs].sum;
    else return query(mid+1,r,x,tree[y].rs);
}

void clean(){
    cnt=0;
    memset(pre,0,sizeof(pre));
    memset(root,0,sizeof(root));
    memset(a,0,sizeof(a));
    memset(tree,0,sizeof(tree));
}

int main(){
    while(scanf("%d%d",&n,&q)!=EOF){
        clean();
        for(int i=1;i<=n;i++) scanf("%d",&a[i]),a[i+n]=a[i];
            for(int i=1;i<=(n<<1);i++){
            if(!pre[a[i]]) update(1,n<<1,root[i],root[i-1],i,1);
            else{
                update(1,n<<1,temp,root[i-1],pre[a[i]],-1);
                update(1,n<<1,root[i],temp,i,1);
            }
            pre[a[i]]=i;
            }
        for(int i=1;i<=q;i++){
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%d\n",query(1,n<<1,r,root[l+n]));
        }
    }
    return 0;
}

 

展开阅读全文

没有更多推荐了,返回首页