SPOJ - DQUERY D-query

练习离散化所以用了f，ref数组。这道题的ai范围是1e6，或许可以不用离散化？ 

 

#include <stdlib.h>
#include <cstdio>
#include <iostream>
#include <queue>
#include <stack>
#include <string.h>
#include <math.h>
#include <algorithm>

#define mem(t, v)   memset ((t) , v, sizeof(t))
#define INF 0x3f3f3f3f
#define eps 1e-8
typedef long long ll;

using namespace std;

const int MAX=100010;
int t,n,m;
int f[MAX];
int ref[MAX];
int mark[MAX];
struct s{
    int val,id;
}in[MAX];
struct z{
    int x,y,id,ans;
}q[MAX*2];
struct node{
    int l,r;
    int sum;
}tree[MAX<<2];

bool cmp(s a,s b){
    return a.val<b.val;
}

bool _cmp(s a,s b){
    return a.id<b.id;
}

bool cmpq(z a,z b){
    return a.y<b.y;
}

bool _cmpq(z a,z b){
    return a.id<b.id;
}

void Build(int i,int l,int r){
    tree[i].l=l,tree[i].r=r,tree[i].sum=0;
    if(l==r)return ;
    int mid=(l+r)>>1;
    Build(i<<1,l,mid);
    Build(i<<1|1,mid+1,r);
}

void Update(int i,int x,int k){
    if(tree[i].l==tree[i].r){
        tree[i].sum+=k;
        return ;
    }
    int mid=(tree[i].r+tree[i].l)>>1;
    if(mid>=x)Update(i<<1,x,k);
    if(mid+1<=x)Update(i<<1|1,x,k);
    tree[i].sum=tree[i<<1].sum+tree[i<<1|1].sum;
}

int Search(int i,int l,int r){
    if(tree[i].l>=l&&tree[i].r<=r){
        return tree[i].sum;
    }
    if(tree[i].l>r||tree[i].r<l)return 0;
    int s=0;
    if(tree[i<<1].r>=l)s+=Search(i<<1,l,r);
    if(tree[i<<1|1].l<=r)s+=Search(i<<1|1,l,r);
    return s;
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&in[i].val);
        in[i].id=i;
    }
    sort(in+1,in+1+n,cmp);
    int top=0;
    in[0].val=-1;
    for(int i=1;i<=n;i++){
        if(in[i].val!=in[i-1].val)
            f[++top]=in[i].val;
        ref[in[i].id]=top;
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++){
        scanf("%d%d",&q[i].x,&q[i].y);
        q[i].id=i;
    }
    sort(q,q+m,cmpq);
    Build(1,1,n);
    mem(mark,0);
    for(int i=0,j=1;i<m;i++){
        while(j<=n&&j<=q[i].y){
            int t=ref[j];
            if(mark[t]){
                Update(1,mark[t],-1);
            }
            Update(1,j,1);
//            printf("%d %d %d+\n",j,t,mark[t]);
            mark[t]=j++;
        }
        q[i].ans=Search(1,q[i].x,q[i].y);
    }
    sort(q,q+m,_cmpq);
    for(int i=0;i<m;i++)
        printf("%d\n",q[i].ans);
    return 0;
}