SPOJ DQUERY (主席樹模板)
阿新 • • 發佈:2019-01-25
題意:
給出一個序列,詢問區間內有多少個不同的數
這題卡分塊莫隊,寫了一下主席樹,已加入模板
主席樹大概是這麼回事,每個結點記錄字首線段樹,當然這裡的線段樹結點的申請是動態的,每次最多申請logn個,對於詢問來說就只需要詢問字首r線段樹中l到n區間內不同數的個數了
#include<cstring> #include<cstdio> #include<iostream> #include<algorithm> #include<map> using namespace std; #define maxn 30004 int a[maxn],root[maxn],tot; struct node { int l,r; int s; node() { l=r=s=0; } } t[30*maxn]; int newnode(int s,int l,int r) { int rt=++tot; t[rt].s=s; t[rt].l=l; t[rt].r=r; return rt; } void update(int &rt,int pre,int pos,int l,int r,int val) { node &temp=t[pre]; rt=newnode(temp.s+val,temp.l,temp.r); if(l==r) return ; int mid=l+r>>1; if(pos<=mid) update(t[rt].l,temp.l,pos,l,mid,val); else update(t[rt].r,temp.r,pos,mid+1,r,val); } int query(int rt,int pos,int l,int r) { if(l==pos) return t[rt].s; int mid=l+r>>1; if(pos<=mid) return query(t[rt].l,pos,l,mid)+t[t[rt].r].s; return query(t[rt].r,pos,mid+1,r); } int main() { int n; while(scanf("%d",&n)!=EOF) { tot=0; for(int i=1; i<=n; i++) { scanf("%d",&a[i]); } map<int ,int >mp; int temp; for(int i=1; i<=n; i++) { int e=a[i]; if(!mp[e]) { update(root[i],root[i-1],i,1,n,1); } else { update(temp,root[i-1],mp[e],1,n,-1); update(root[i],temp,i,1,n,1); } mp[e]=i; } int q; scanf("%d",&q); while(q--) { int l,r; scanf("%d%d",&l,&r); printf("%d\n",query(root[r],l,1,n)); } } return 0; }