查询区间内有多少个不同的数（线段树/树状数组）

入门级数据结构算法。复习一下，分别手写一个。

线段树版本（过了CF上的https://codeforces.com/contest/1291/problem/D）：

#include<bits/stdc++.h>
#define f(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
struct qujian{
    int l,r,index;
    friend bool operator < (const qujian q1,const qujian q2){
        return q1.r<q2.r;
    }
}qu[100005];
int q;

const int N=2e5+5;
int a[N];
struct node{
    int l,r,val;
}tree[N<<2];
void build(int rt,int l,int r){
    tree[rt].l=l,tree[rt].r=r;
    if(l==r){
        tree[rt].val=0;
        return;
    }
    int mid=l+r>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    tree[rt].val=tree[rt<<1].val+tree[rt<<1|1].val;
}
void update(int rt,int pos,int val){
    if(tree[rt].l==tree[rt].r&&pos==tree[rt].l){
        tree[rt].val+=val;
        return;
    }
    int mid=tree[rt].l+tree[rt].r>>1;
    if(pos<=mid) update(rt<<1,pos,val);
    else update(rt<<1|1,pos,val);
    tree[rt].val+=val;
}
int query(int rt,int l,int r){
    if(l<=tree[rt].l&&r>=tree[rt].r) return tree[rt].val;
    int mid=tree[rt].l+tree[rt].r>>1;
    if(l>mid) return query(rt<<1|1,l,r);
    else if(r<=mid) return query(rt<<1,l,r);
    else return query(rt<<1,l,mid)+query(rt<<1|1,mid+1,r);
}

map<char,int> mp;
char b[N];
int ans[N];

bool cmp(const qujian q1,const qujian q2){
    return q1.index<q2.index;
}

int main(){
    scanf("%s",b+1);
    int len=strlen(b+1);
    scanf("%d",&q);
    f(i,1,q) scanf("%d%d",&qu[i].l,&qu[i].r),qu[i].index=i;
    sort(qu+1,qu+1+q);
    //f(i,1,q) cout<<qu[i].l<<" "<<qu[i].r<<endl;
    int cur=1;
    build(1,1,len);
    f(i,1,q){
        f(j,cur,qu[i].r){
            //cout<<"the current qujian is "<<qu[i].l<<" "<<qu[i].r<<endl;
            int p=mp[b[j]];
            if(p!=0){
                update(1,p,-1);
                //cout<<b[j]<<" has appeared at "<<p<<endl;
            }
            update(1,j,1);
            //cout<<b[j]<<" last appeared at "<<j<<endl;
            //cout<<"now the "<<j<<" is "<<query(1,j,j)<<endl;
            mp[b[j]]=j;
        }
        cur=qu[i].r+1;
        ans[qu[i].index]=query(1,qu[i].l,qu[i].r);
        //cout<<qu[i].index<<":["<<qu[i].l<<","<<qu[i].r<<"]="<<query(1,qu[i].l,qu[i].r)<<endl;
    }
    //f(i,1,q) cout<<ans[i]<<" ";
    sort(qu+1,qu+1+q,cmp);
    f(i,1,q){
        if(qu[i].r-qu[i].l+1==1){
            puts("YES");
        }
        else if(ans[i]>=3){
            puts("YES");
        }
        else if(b[qu[i].l]!=b[qu[i].r]){
            puts("YES");
        }
        else puts("NO");
    }
} 

树状数组版本（怪不得潘神这么喜欢，写起来确实短得多）：

#include<bits/stdc++.h>
#define f(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
struct qujian{
    int l,r,index;
    friend bool operator < (const qujian q1,const qujian q2){
        return q1.r<q2.r;
    }
}qu[100005];
bool cmp(const qujian q1,const qujian q2){
    return q1.index<q2.index;
}
const int N=1e5+5;
map<int,int> mp;
int ans[N];
int len[N];
int n,q;
int a[N];
int tree[N];
#define lowbit(x) (x&(-x))
void update(int pos,int val){
    while(pos<=n){
        tree[pos]+=val;
        pos+=lowbit(pos);
    }
}
int sum_1_to_pos(int pos){
    int ans=0;
    while(pos>0){
        ans+=tree[pos];
        pos-=lowbit(pos);
    }
    return ans;
}
#define query(x,y) (sum_1_to_pos(y)-sum_1_to_pos(x-1))

int main(){
    scanf("%d%d",&n,&q);
    f(i,1,n) scanf("%d",&a[i]);
    f(i,1,q) scanf("%d%d",&qu[i].l,&qu[i].r),qu[i].index=i,len[i]=qu[i].r-qu[i].l+1;
    sort(qu+1,qu+1+q);
    int cur=1;
    f(i,1,q){
        f(j,cur,qu[i].r){
            int p=mp[a[j]];
            if(p!=0){
                update(p,-1);
            }
            update(j,1);
            mp[a[j]]=j;
        }
        cur=qu[i].r+1;
        ans[qu[i].index]=query(qu[i].l,qu[i].r);
    }
    f(i,1,q){
        if(ans[i]==len[i]) puts("Yes");
        else puts("No");
    }
}