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

最新推荐文章于 2022-05-29 15:09:12 发布

窦小雨

最新推荐文章于 2022-05-29 15:09:12 发布

阅读量325

点赞数

分类专栏：数据结构—树状数组文章标签： BZOJ

本文链接：https://blog.csdn.net/qq_33402621/article/details/53169900

版权

数据结构—树状数组专栏收录该内容

7 篇文章 0 订阅

订阅专栏

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];
int	lowbit(int	x)
{
    return	x&(-x);
}
void add(int x,int c)
{
    while(x<=MAXN)
    {
        a[x]+=c;
        x+=lowbit(x);
    }
}
int n,m;

int	get_sum(int	x)
{
    int	ret=0;
    while(x>0)
    {
        ret+=a[x];
        x-=lowbit(x);
    }
    return	ret;
}
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;
}