【SDOI2009】【BZOJ1878】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

Day2

其实我以前从来没集中刷过线段树,所以这题也一直没做过

考虑离线处理,前缀和作差得答案.直接BIT维护一下就行了.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define lowbit(x) (x & (- x))
#define MAXN 1000010
#define GET (ch >= '0' && ch <= '9')
using namespace std;
int n,m;
int pos[MAXN];
int c[50010];
int a[50010],next[50010];
void add(int x,int delta)
{
    for (int i = x;i <= n;i += lowbit(i))   c[i] += delta;
}
int query(int x)
{
    int ret = 0;
    for (int i = x;i;i -= lowbit(i))    ret += c[i];
    return ret;
}
struct Query
{
    int l,r,id,ans;
    bool operator <(const Query& a)const
    {
        return l == a.l ? r < a.r : l < a.l;
    }
}q[200010];
void in(int &x)
{
    char ch = getchar();x = 0;
    while (!GET)    ch = getchar();
    while (GET) x = x * 10 + ch - '0',ch = getchar();
}
bool cmp(Query a,Query b)
{
    return a.id < b.id;
}
int main()
{
    in(n);int maxn = 0;
    for (int i = 1;i <= n;i++)  in(a[i]),maxn = max(maxn,a[i]);
    for (int i = n;i;i--)   next[i] = pos[a[i]],pos[a[i]] = i;
    for (int i = 1;i <= maxn;i++)
        if (pos[i]) add(pos[i],1);
    in(m);int now = 1;
    for (int i = 1;i <= m;i++)  in(q[i].l),in(q[i].r),q[i].id = i;
    sort(q + 1,q + m + 1);
    for (int i = 1;i <= m;i++)
    {
        while (now < q[i].l)
        {
            if (next[now])  add(next[now],1);
            now++;
        }
        q[i].ans = query(q[i].r) - query(q[i].l - 1);
    }
    sort(q + 1,q + m + 1,cmp);
    for (int i = 1;i <= m;i++)  printf("%d\n",q[i].ans);
}

你可能感兴趣的:(树状数组)