SPOJ D-query(区间不同的数的个数)

题目

离线：树状数组

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=(int)3e4+10;
struct node{
int l,r,idx;
}qu[200010];
bool cmp(const node &a,const node &b){
return a.r<b.r;
}
vector<int>v;
int a[MAXN],pre[MAXN],mark[MAXN];
int ans[200010];
int p[MAXN];
int n;
void add(int x,int y){
while(x<=n){
p[x]+=y;
x+=x&(-x);
}
}
int getsum(int x){
ll re=0;
while(x){
re+=p[x];
x-=x&(-x);
}
return re;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
v.push_back(a[i]);
}
sort(v.begin(),v.end());
for(int i=1;i<=n;i++){
a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
pre[i]=mark[a[i]];
mark[a[i]]=i;
}
int m;
scanf("%d",&m);
for(int i=1;i<=m;i++){
scanf("%d%d",&qu[i].l,&qu[i].r);
qu[i].idx=i;
}
sort(qu+1,qu+1+m,cmp);
int tmp=0;
for(int i=1;i<=m;i++){
while(tmp<qu[i].r){
tmp++;
if(pre[tmp]==0){
add(1,1);
add(tmp+1,-1);
}
else {
add(pre[tmp]+1,1);
add(tmp+1,-1);
}
}
ans[qu[i].idx]=getsum(qu[i].l);
}
for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
return 0;
}

在线：主席树
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int MAXN=30005;
int a[MAXN],n,m,vsz,pre[MAXN];
vector<int>v;
int root[MAXN],ls[MAXN*40],rs[MAXN*40],add[MAXN*40],tot;
void build(int l,int r,int &rt){
rt=++tot;
if(l==r)return;
int mid=l+r>>1;
build(l,mid,ls[rt]);
build(mid+1,r,rs[rt]);
}
void insert(int l,int r,int x,int y,int &r1,int r2){
r1=++tot;ls[r1]=ls[r2];rs[r1]=rs[r2];add[r1]=add[r2];
if(l==x&&r==y){add[r1]++;return ;}
int mid=l+r>>1;
if(mid>=y)insert(l,mid,x,y,ls[r1],ls[r2]);
else if(mid<x)insert(mid+1,r,x,y,rs[r1],rs[r2]);
else insert(l,mid,x,mid,ls[r1],ls[r2]),insert(mid+1,r,mid+1,y,rs[r1],rs[r2]);
}
int query(int l,int r,int x,int rt){
if(l==r)return add[rt];
int mid=l+r>>1;
if(x<=mid)return add[rt]+query(l,mid,x,ls[rt]);
else return add[rt]+query(mid+1,r,x,rs[rt]);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
v.push_back(a[i]);
}
sort(v.begin(),v.end());
v.erase(unique(v.begin(),v.end()),v.end());
vsz=v.size();
build(1,n,root[0]);
for(int i=1;i<=n;i++){
a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
insert(1,n,pre[a[i]]+1,i,root[i],root[i-1]);
pre[a[i]]=i;
}
scanf("%d",&m);
while(m--){
int l,r;
scanf("%d%d",&l,&r);
printf("%d\n",query(1,n,l,root[r]));
}
return 0;
}