题意:给一个数列,然后多次询问,询问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;
}