Light OJ 1188 Fast Queries （树状数组离线）

题目链接：

http://lightoj.com/volume_showproblem.php?problem=1188

题意：查找一个区间内不同数字的个数。

分析：

离线做法。先按r排序，标记每个数前面出现的位置。处理到位置i时，如果a[i]在前面出现过，那么把前面那个位置在树状数组中清0。然后标记这个数的新位置。然后就是求区间问题了。

<span style="color:#009900;">#include <iostream>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <string.h>
#include <map>
#include <set>
using namespace std;
#define MAXN 100005
#define LL long long
#define INF 0x3f7f7f7f
int a[MAXN],c[MAXN];
int ans[MAXN],vis[MAXN],pre[MAXN];
struct node
{
    int l,r,id;
}p[MAXN];
bool cmp(node x,node y)
{
    if(x.r==y.r)return x.l<y.l;
    return x.r<y.r;
}
int n,q;
void add(int x,int val)
{
    while(x<=n)
    {
        c[x]+=val;
        x+=(-x)&x;;
    }
}
int query(int x)
{
    int sum=0;
    while(x>0)
    {
        sum+=c[x];
        x-=(-x)&x;
    }
    return sum;
}
int main()
{
    int i,j,t,tt=0;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&q);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            vis[a[i]]=c[i]=0;
        }
        for(i=1;i<=q;i++)
        {
            scanf("%d%d",&p[i].l,&p[i].r);
            p[i].id=i;
            ans[i]=0;
        }
        p[q+1].l=p[q+1].r=INF;
        sort(p+1,p+1+q,cmp);
        j=1;

        for(i=1;i<=n;)
        {
            int flag=0;
            while(i<=p[j].r)
            {
                if(p[j].r==INF)
                {
                    flag=1;break;
                }
                if(vis[a[i]])
                {
                    add(pre[a[i]],-1);
                    add(i,1);
                    pre[a[i]]=i;
                }
                else
                {
                    vis[a[i]]=1;
                    add(i,1);
                    pre[a[i]]=i;
                }
                while(i==p[j].r)
                {
                    ans[p[j].id]=query(p[j].r)-query(p[j].l-1);
                    j++;
                }
                i++;
                if(i>n)break;
            }
            if(flag==1)break;
        }
        printf("Case %d:\n",++tt);
        for(i=1;i<=q;i++)
        {
          printf("%d\n",ans[i]);
        }
    }
}</span><span style="color:#ff0000;">
</span>