SPOJ 3267 主席树||线段树+离线

点击打开链接

题意：给一个数列，然后多次询问，询问l到r区间内不同数字的个数

思路：主席树模版题目或者线段树+离线操作都可以，而线段树+离线的话HDU 3333 也是一道基础的题目，求得是不同数字的和，变成个数就是这道题目了

先是主席树的，代码参考kuangbin巨巨

#include <map>
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=30010;
int T[maxn*30],lson[maxn*30],rson[maxn*30],c[maxn*30],a[maxn];
int n,q,tot;
int buildtree(int le,int ri){
    int root=tot++;
    c[root]=0;
    if(le!=ri){
        int mid=(le+ri)>>1;
        lson[root]=buildtree(le,mid);
        rson[root]=buildtree(mid+1,ri);
    }
    return root;
}
int update(int root,int pos,int val){
    int newroot=tot++,tmp=newroot;
    c[newroot]=c[root]+val;
    int l=1,r=n;
    while(l<r){
        int mid=(l+r)>>1;
        if(pos<=mid){
            lson[newroot]=tot++;rson[newroot]=rson[root];
            newroot=lson[newroot];root=lson[root];
            r=mid;
        }else{
            rson[newroot]=tot++;lson[newroot]=lson[root];
            newroot=rson[newroot];root=rson[root];
            l=mid+1;
        }
        c[newroot]=c[root]+val;
    }
    return tmp;
}
int query(int root,int pos){
    int ret=0,l=1,r=n;
    while(pos<r){
        int mid=(l+r)>>1;
        if(pos<=mid){
            r=mid;root=lson[root];
        }else{
            ret+=c[lson[root]];
            root=rson[root];
            l=mid+1;
        }
    }
    return ret+=c[root];
}
int main(){
    int u,v,k;
    while(scanf("%d",&n)!=-1){
        tot=0;
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        T[n+1]=buildtree(1,n);
        map<int,int>M;
        for(int i=n;i>=1;i--){
            if(M.find(a[i])==M.end()) T[i]=update(T[i+1],i,1);
            else{
                int pos=update(T[i+1],M[a[i]],-1);
                T[i]=update(pos,i,1);
            }
            M[a[i]]=i;
        }
        scanf("%d",&q);
        while(q--){
            scanf("%d%d",&u,&v);
            printf("%d\n",query(T[u],v));
        }
    }
    return 0;
}

离线+线段树   或者离线+树状数组一样  代码自己写的有点丑

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=200010;
ll num[maxn<<2];
void update(int pos,int val,int le,int ri,int node){
    if(le==ri){
        num[node]=val;
        return ;
    }
    int t=(le+ri)>>1;
    if(pos<=t) update(pos,val,le,t,node<<1);
    else update(pos,val,t+1,ri,node<<1|1);
    num[node]=num[node<<1]+num[node<<1|1];
}
ll query(int l,int r,int le,int ri,int node){
    if(l<=le&&ri<=r) return num[node];
    int t=(le+ri)>>1;
    ll ans=0;
    if(l<=t) ans+=query(l,r,le,t,node<<1);
    if(r>t) ans+=query(l,r,t+1,ri,node<<1|1);
    return ans;
}
struct edge{
    int num,num1,pos;
}B[maxn*4];
bool cmp2(const edge &a,const edge &b){
    return a.num1<b.num1;
}
int vis[maxn],tmp[maxn],pre[maxn],A[maxn];
int ans[maxn*4];
int main(){
    int T,n,m,a,b;
    while(scanf("%d",&n)!=-1){
        memset(vis,0,sizeof(vis));
        memset(ans,0,sizeof(ans));
        memset(pre,-1,sizeof(pre));
        memset(num,0,sizeof(num));
        for(int i=0;i<n;i++){
            scanf("%d",&A[i]);
            tmp[i]=A[i];
        }
        sort(tmp,tmp+n);
        scanf("%d",&m);
        for(int i=0;i<m;i++){
            scanf("%d%d",&B[i].num,&B[i].num1);
            B[i].pos=i;
        }
        sort(B,B+m,cmp2);
        for(int i=0,j=0;i<n;i++){
            int t=lower_bound(tmp,tmp+n,A[i])-tmp;
            if(vis[t]){
                update(pre[t],0,1,n,1);
                update(i+1,1,1,n,1);
                pre[t]=i+1;
            }else{
                vis[t]=1;
                update(i+1,1,1,n,1);
                pre[t]=i+1;
            }
            for(;j<m;j++){
                if(i+1!=B[j].num1) break;
                ans[B[j].pos]=query(B[j].num,B[j].num1,1,n,1);
            }
        }
        for(int i=0;i<m;i++) printf("%d\n",ans[i]);
    }
    return 0;
}

第二种模版写法：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=30010;
int sum[maxn*50],L[maxn*50],Root[maxn*2],R[maxn*50];
int n,q,cnt,croot;
int ma[maxn],last[1000010];
void buildtree(int le,int ri,int node){
    if(le==ri){
        sum[node]=0;
        return ;
    }
    int t=(le+ri)>>1;
    L[node]=++cnt;buildtree(le,t,cnt);
    R[node]=++cnt;buildtree(t+1,ri,cnt);
    sum[node]=sum[L[node]]+sum[R[node]];
}
void update(int pos,int val,int pre,int le,int ri,int node){
    if(le==ri){
        sum[node]=sum[pre]+val;
        return ;
    }
    int t=(le+ri)>>1;
    if(pos<=t){
        L[node]=++cnt;
        update(pos,val,L[pre],le,t,cnt);
        R[node]=R[pre];
    }else{
        L[node]=L[pre];
        R[node]=++cnt;
        update(pos,val,R[pre],t+1,ri,cnt);
    }
    sum[node]=sum[L[node]]+sum[R[node]];
}
int query(int r,int le,int ri,int node){
    if(ri<=r) return sum[node];
    int t=(le+ri)>>1;
    if(r<=t) return query(r,le,t,L[node]);
    else return sum[L[node]]+query(r,t+1,ri,R[node]);
}
int main(){
    int x,y;
    while(scanf("%d",&n)!=-1){
        cnt=0;croot=0;
        Root[++croot]=++cnt;ma[0]=1;
        buildtree(1,n,cnt);
        for(int i=1;i<=n;i++){
            scanf("%d",&x);
            Root[++croot]=++cnt;
            update(i,1,Root[croot-1],1,n,cnt);
            if(last[x]){
                Root[++croot]=++cnt;
                update(last[x],-1,Root[croot-1],1,n,cnt);
            }
            last[x]=i;ma[i]=Root[croot];
        }
        scanf("%d",&q);
        while(q--){
            scanf("%d%d",&x,&y);
            if(x>1) printf("%d\n",sum[ma[y]]-query(x-1,1,n,ma[y]));
            else printf("%d\n",sum[ma[y]]);
        }
    }
    return 0;
}