SPOJ-DQUERY - D-query(主席数,区间不同元素个数)

题意:多次询问一段区间中不同元素的个数
思路:主席树入门题,建N棵线段树,每颗记录区间[1,n]区间中每个元素最后出现的位置。查询的时候直接相减。

#include 
#include 
using namespace std;
const int maxn=1e5+100;
int a[maxn],root[maxn],f[maxn],cur,lastnum[maxn];
struct node{
    int cnt,flag;
    int left,right;
}son[maxn*20];
void pushup(int x){
    son[x].cnt=son[son[x].right].cnt+son[son[x].left].cnt;
}
int build(int l,int r){
    int k=cur++;
    if(l==r){
        son[k].cnt=0;
        son[k].flag=0;
        return k;
    }
    int mid=(l+r)/2;
    son[k].left=build(l,mid);
    son[k].right=build(mid+1,r);
    pushup(k);
    return k;
}
int update(int x,int l,int r,int pos,int val){
    int k=cur++;
    son[k]=son[x];
    if(l==r){
        son[k].cnt+=val;
        return k;
    }
    int mid=(l+r)/2;
    if(pos<=mid) son[k].left=update(son[x].left,l,mid,pos,val);
    else son[k].right=update(son[x].right,mid+1,r,pos,val);
    pushup(k);
    return k;
}
int query(int a,int b,int l,int r,int x,int y){
    if(l>=x&&r<=y)return son[b].cnt-son[a].cnt;
    int mid=(l+r)/2;
    if(mid>=y)return query(son[a].left,son[b].left,l,mid,x,y);
    else if(midreturn query(son[a].right,son[b].right,mid+1,r,x,y);
    else return query(son[a].left,son[b].left,l,mid,x,mid)+query(son[a].right,son[b].right,mid+1,r,mid+1,y);
}
int main(){
    int n,m;
    cur=1;
    scanf("%d",&n);
        for (int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            f[i] = a[i];
        }
        int sum = 1;
        sort(f + 1, f + 1 + n);
        for (int i = 2; i <= n; i++) {//数据离散化
            if (f[i] != f[i - 1])f[++sum] = f[i];
        }
        int temproot=0;
        for (int i = 1; i <= n; i++) {//每多加一个数,只记录这个数最后出现的位置
            a[i] = lower_bound(f + 1, f + 1 + sum, a[i]) - f;
            if(!lastnum[a[i]]){//如果这个数不是第一次出现,把这个数前一次出现位置的标记删除,再在当前位置标记
                root[i]=update(root[i-1],1,n,i,1);
            }else{
                temproot=update(root[i-1],1,n,lastnum[a[i]],-1);
                root[i]=update(temproot,1,n,i,1);
            }
            lastnum[a[i]]=i;
        }
        scanf("%d",&m);
        for (int i = 0; i < m; i++) {
            int a, b;
            scanf("%d%d", &a, &b);
            printf("%d\n",query(root[a-1],root[b],1,n,a,b));//每次查询,用区间[1,r]得标记数减去[1,l-1]区间的标技数
        }
    return 0;
}

你可能感兴趣的:(C++,数据结构,可持续化线段树)