1878: [SDOI2009]HH的项链 树状数组

1878: [SDOI2009]HH的项链

Time Limit: 4 Sec  Memory Limit: 64 MB
Submit: 3486  Solved: 1738
[Submit][Status][Discuss]

Description

HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运，所以每次散步 完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH不断地收集新的贝壳，因此， 他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同 的贝壳？这个问题很难回答。。。因为项链实在是太长了。于是，他只好求助睿智的你，来解 决这个问题。

Input

第一行：一个整数N，表示项链的长度。 第二行：N个整数，表示依次表示项链中贝壳的编号（编号为0到1000000之间的整数）。 第三行：一个整数M，表示HH询问的个数。 接下来M行：每行两个整数，L和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

Output

M行，每行一个整数，依次表示询问对应的答案。

Sample Input

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

Sample Output

2
2
4

HINT

对于20%的数据，N ≤ 100，M ≤ 1000；
对于40%的数据，N ≤ 3000，M ≤ 200000；
对于100%的数据，N ≤ 50000，M ≤ 200000。

Source

题目大意：

如题。

思路：

首先进行预处理。

nextt [ i ] 代表和 i 相同颜色的下一个位置。

pre [ i ] 代表和  i 颜色相同的上一个位置。

然后对于m 个查询，我们按照左端点进行排序。

对于树状数组  a ，我们仅将某个颜色第一次出现的位置赋值 为 1 ，其余的赋值为 0 。这样的话我们在查询

[ 1  r  ]   区间内颜色数量的时候 直接 sum （i）就是答案。

那么我们在查找 [ 2 , r ] 的时候就是将nextt [ 1 ] 位置的改为 1    第一个位置改为 0 

然后结果就出来了。

stl 有毒  我以为map  很厉害   结果开了一个 百万的数组比map 快多了。

还有一个很不解的问题，为什么  输出的时候  我用printf 可以。用cout 就  RE  谁来给我解释一下  RE 是个什么鬼？

AC代码：

#include<iostream>#include<algorithm>#include<string.h>#include<stdio.h>#include<map>using namespace std;const int MAXN=50005;int a[MAXN];int num[MAXN];int nextt[MAXN];int pre[MAXN];int hehe[MAXN];int ans[MAXN];int mm[1000005];struct node{    int ss;    int ee;    int pos;    friend bool operator < (const node &aa,const node &bb)    {        return aa.ss<bb.ss;    }}cc[200005];intlowbit(intx){    returnx&(-x);}void add(int x,int c){    while(x<=MAXN)    {        a[x]+=c;        x+=lowbit(x);    }}int n,m;intget_sum(intx){    intret=0;    while(x>0)    {        ret+=a[x];        x-=lowbit(x);    }    returnret;}void init(){    memset(a,0,sizeof(a));    memset(hehe,0,sizeof(hehe));    memset(nextt,-1,sizeof(nextt));    memset(mm,0,sizeof(mm));    for(int i=1;i<=n;i++)    {        scanf("%d",&num[i]);    }    int cnt=1;    for(int i=1;i<=n;i++)    {        if(mm[num[i]]==0)        {            mm[num[i]]=cnt;            cnt++;        }    }    for(int i=1;i<=n;i++)    {        if(hehe[mm[num[i]]]==0)        {            hehe[mm[num[i]]]++;            add(i,1);            pre[mm[num[i]]]=i;        }        else        {            nextt[pre[mm[num[i]]]]=i;            pre[mm[num[i]]]=i;        }    }    scanf("%d",&m);    for(int i=0;i<m;i++)    {        scanf("%d%d",&cc[i].ss,&cc[i].ee);        cc[i].pos=i+1;    }    stable_sort(cc,cc+m);}int main(){    while(~scanf("%d",&n))    {        init();        int pp=1;        for(int i=0;i<m;i++)        {            for(int j=pp;j<cc[i].ss;j++)            {                add(j,-1);                if(nextt[j]!=-1)                    add(nextt[j],1);            }            pp=cc[i].ss;            ans[cc[i].pos]=get_sum(cc[i].ee);        }        for(int i=1;i<=m;i++)        {            printf("%d\n",ans[i]);        }    }    return 0;}