【洛谷P1972】[SDOI2009]HH的项链

题目描述

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

输入输出格式

输入格式：

第一行：一个整数N，表示项链的长度。

第二行：N 个整数，表示依次表示项链中贝壳的编号（编号为0 到1000000 之间的整数）。

第三行：一个整数M，表示HH 询问的个数。

接下来M 行：每行两个整数，L 和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

输出格式：

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

输入输出样例

输入样例#1：

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

输出样例#1：

2
2
4

说明

数据范围：

对于100%的数据，N <= 50000，M <= 200000。

题解：

先把所有的区间按照左端点排序，从左到右扫。

对于区间左面的点，因为已经没有用了，那么它的下一个和它颜色相同的点个数加1。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=50000+5;
const int maxm=200000+5;
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,mx,x=1;
int a[maxn],c[maxn],nxt[maxn],head[1000005],ans[maxm];
struct node
{
    int x,y,id;
    bool operator < (const node& j) const {
        return x<j.x;
    }
}q[maxm];
inline void update(int x,int k)
{
    for( ;x<=n;x+=x&(-x)) c[x]+=k;
}
inline int getsum(int x)
{
    int t=0;
    for( ;x;x-=x&(-x)) t+=c[x];
    return t;
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=n;i>=1;i--)
    {
        nxt[i]=head[a[i]];
        head[a[i]]=i;
        mx=max(mx,a[i]);
    }
    for(int i=1;i<=mx;i++)
    if(head[i]) update(head[i],1);
    m=read();
    for(int i=1;i<=m;i++)
    {    
        q[i].x=read();q[i].y=read();
        q[i].id=i;
    }
    sort(q+1,q+m+1);
    for(int i=1;i<=m;i++)
    {
        while(x<q[i].x)
        {
            if(nxt[x]) update(nxt[x],1);
            x++;
        }
        ans[q[i].id]=getsum(q[i].y)-getsum(q[i].x-1);
    }
    for(int i=1;i<=m;i++)
        printf("%d
",ans[i]);
    return 0;
}