【BZOJ1878】[SDOI2009]HH的项链 离线BIT

时间：2022-06-04 00:23:23

1878: [SDOI2009]HH的项链

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

题解：

　　我们考虑离线算法

　　首先记录下每种颜色的下一种颜色所在的位置

　　将所有询问按照左端点进行排序

　　将所有颜色的第一个点x a[x]++

　　然后从左往右扫

　　扫到一个点x将a[next[x]]++

　　碰到一个询问l,r输出sum[r]-sum[l-1]

　　其中sum是a数组的前缀和

　　求前缀和可以用树状数组默默@HZWER

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<cstring>

using namespace std;

inline int read()

{

    int x=;char ch=getchar();

    while(ch<''||ch>''){ch=getchar();}

    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}

    return x;

}

int n,m,mx;

int a[],next[],t[];

int p[];

struct data{int l,r,id,ans;}q[];

bool cmp1(data a,data b)

{return a.l==b.l?a.r<b.r:a.l<b.l;}

bool cmp2(data a,data b)

{return a.id<b.id;}

int lowbit(int x){return x&(-x);}

void update(int x,int v)

{

    for(int i=x;i<=n;i+=lowbit(i))

        t[i]+=v;

}

int ask(int x)

{

    int tmp=;

    for(int i=x;i>;i-=lowbit(i))

        tmp+=t[i];

    return tmp;

}

int main()

{

    n=read();

    for(int i=;i<=n;i++)

        a[i]=read(),mx=max(mx,a[i]);

    for(int i=n;i>;i--)

        next[i]=p[a[i]],p[a[i]]=i;

    for(int i=;i<=mx;i++)

        if(p[i])update(p[i],);

       // cout<<ask(4)<<endl;

    m=read();

    for(int i=;i<=m;i++)

        q[i].l=read(),q[i].r=read(),q[i].id=i;

    sort(q+,q+m+,cmp1);

    int l=;

    for(int i=;i<=m;i++)

    {

        while(l<q[i].l)

        {

            if(next[l])update(next[l],);

            //cout<<next[l]<<endl;

            l++;

        }

        q[i].ans=ask(q[i].r)-ask(q[i].l-);

    }

    sort(q+,q+m+,cmp2);

    for(int i=;i<=m;i++)

        printf("%d\n",q[i].ans);

    return ;

}

代码

相关文章