Hrbust 1792 Number Squence 【思维+离线+离散化BIT】

Number Squence

Time Limit: 1000 MS Memory Limit: 32768 K Total Submit: 32(10 users) Total Accepted: 7(7 users) Rating:  Special Judge: No

Description

Ikki最近又喜欢上了数列游戏，她现在想出个问题考考你，问题是这样的：     首先给你一个由n个整数组成的序列然后Ikki会给出一系列的提问：知道了区间的两个端点s和e，那么这个区间内有多少个不同的数字？

Input

多组测试数据，处理到文件结束，对于每组数据： 第一行输入两个整数n，q分别表示序列中数的个数和Ikki提问的次数 （0＜n≤100000,0<q≤100000）。 第二行输入n个整数xi~xn表示序列中每个数的大小。(0 <= x <= 109 ） 接下来的q行每行输入两个正整数a，b(a≤b)表示询问的区间为[a,b]。（序列下标从1开始）

Output

对于每组测试数据，输出q行依次表示每个提问的答案。

Sample Input

6 3 1 2 1 2 3 4 1 2 1 3 2 5

Sample Output

2 2 3

Author

周洲 @hrbust

思路：

 可以用主席树来写->_->

我们首先将每个数字处理成一条链。nex【i】表示下一个数字a【i】的位子。

那么我们这里逆序处理一下即可。

然后我们将询问离线，按照询问区间左端点从小到大排序。

那么我们初始的时候，可以将所有数字都入树，对于数字a【i】，如果他是在数组中从左到右第一个出现的，那么对应update（i，1）；

因为此时我们遍历查询是按照左端点递增的顺序来的，所以我们只要不断的更新维护当前询问以左的部分即可。

那么我们需要在每一次查询之前，将小于当前左区间的位子上的数删除掉并且使得nex[i]入树，即update（nex[i]，1）；

查询的时候，只要查询q【i】.r以左一共还有多少个数在树上即可。

这部分文字描述可能稍微有些模糊，不妨参考代码，代码实现很简单。

Ac代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<map>
using namespace std;
struct node
{
    int l,r,pos;
}q[120050];
int num[120050];
int nex[120050];
int last[120050];
int a[120050];
int tree[120005];//树
int ans[120005];
int n,qq;
int cmp(node a,node b)
{
    if(a.l==b.l)return a.r<b.r;
    else return a.l<b.l;
}
int lowbit(int x)//lowbit
{
    return x&(-x);
}
int sum(int x)//求和求的是比当前数小的数字之和，至于这里如何实现，很简单：int sum=sum(a[i])；
{
    int sum=0;
    while(x>0)
    {
        sum+=tree[x];
        x-=lowbit(x);
    }
    return sum;
}
void update(int x,int c)//加数据。
{
    while(x<=n)
    {
        tree[x]+=c;
        x+=lowbit(x);
    }
}
int main()
{
    while(~scanf("%d%d",&n,&qq))
    {
        int cnt=0;
        map<int ,int >s;
        memset(ans,0,sizeof(ans));
        memset(num,0,sizeof(num));
        memset(tree,0,sizeof(tree));
        memset(nex,-1,sizeof(nex));
        memset(last,-1,sizeof(last));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            if(s[a[i]]==0)s[a[i]]=++cnt;
            num[i]=s[a[i]];
        }
        for(int i=n;i>=1;i--)
        {
            nex[i]=last[num[i]];
            last[num[i]]=i;
        }
        for(int i=1;i<=cnt;i++)update(last[i],1);
        for(int i=0;i<qq;i++)scanf("%d%d",&q[i].l,&q[i].r),q[i].pos=i;
        sort(q,q+qq,cmp);
        int now=1;
        for(int i=0;i<qq;i++)
        {
            while(now<q[i].l)
            {
                update(now,-1);
                if(nex[now]!=-1)update(nex[now],1);
                now++;
            }
            ans[q[i].pos]=sum(q[i].r);
        }
        for(int i=0;i<qq;i++)printf("%d\n",ans[i]);
    }
}