DQUERY D-query [主席樹]
阿新 • • 發佈:2019-01-23
題意:給出長度為n的數列,m個詢問,每次詢問區間【L,R】中不同數的個數
題解:這題與以往的主席樹不同,用位置建樹,第i個線段樹維護前i個位置的資訊每次向後移動位置分為兩種情況
1、該位置的數之前沒有出現過,則在第i個線段樹的i這個位置加1
2、該位置上的數出現過,則在第i個線段樹上,前一次出現的位置處(設為k)
AC程式碼:
#include<stdio.h> #include<map> using namespace std; #define N 30005 int a[N],tot; int tree[N*25],lchild[N*25],rchild[N*25],root[N*25]; map<int,int>mp; void update(int last,int cur,int L,int R,int x,int k) { tree[cur]=tree[last]; lchild[cur]=lchild[last]; rchild[cur]=rchild[last]; if(L==R) { tree[cur]+=k; return ; } int mid=L+R>>1; if(x<=mid)update(lchild[last],lchild[cur]=++tot,L,mid,x,k); else update(rchild[last],rchild[cur]=++tot,mid+1,R,x,k); tree[cur]=tree[lchild[cur]]+tree[rchild[cur]]; } int query(int root,int l,int r,int L,int R) { if(l<=L&&R<=r) return tree[root]; int mid=L+R>>1; if(r<=mid)return query(lchild[root],l,r,L,mid); else if(mid<l)return query(rchild[root],l,r,mid+1,R); else return query(lchild[root],l,mid,L,mid)+query(rchild[root],mid+1,r,mid+1,R); } int main() { int n; scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=n;i++) { if(!mp[a[i]]) update(root[i-1],root[i]=++tot,1,n,i,1); else { update(root[i-1],root[i]=++tot,1,n,i,1); int now=++tot; update(root[i],now,1,n,mp[a[i]],-1); root[i]=now; } mp[a[i]]=i; } int q; scanf("%d",&q); while(q--) { int l,r; scanf("%d%d",&l,&r); printf("%d\n",query(root[r],l,r,1,n)); } }