bzoj千题计划181：bzoj1878: [SDOI2009]HH的项链

http://www.lydsy.com/JudgeOnline/problem.php?id=1878

之前用莫队做的，现在用树状数组

把每种数的第一个出现位置在树状数组中+1

nxt[i] 记录i后面第一个数字和i相同的位置

对于每一个询问[l,r]，输出[1,r]内数的种类-只在[1,l-1]内数的种类

[1,r]内数的种类就是记录的 每种数的第一个出现位置

只在[1,l-1]内数的种类：

对于<l的i，对应的nxt[i]在树状数组中+1,

这样query（r）-query（l-1）时，

只在[1，l-1]内的被减走

既在[1,l-1] 又在[l,r] 内的减1又加1

#include<cstdio>
#include<iostream>
#include<algorithm>

using namespace std;

#define N 50001
#define M 200001
#define K 1000001

int n;
int a[N];
struct node
{
    int l,r;
    int id;
}e[M];

bool have[K];

int nxt[N],last[K];

int c[N+1];

int ans[M];

#define lowbit(x) x&-x

void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*10+c-'0'; c=getchar(); }
}

bool cmp(node p,node q)
{
    return p.l<q.l;
}

void add(int x)
{
    while(x<=n+1)
    {
        c[x]++;
        x+=lowbit(x);
    }
}

int query(int x)
{
    int sum=0;
    while(x)
    {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}

int main()
{
    read(n);
    for(int i=1;i<=n;++i) 
    {
        read(a[i]);
        if(!have[a[i]])
        {
            have[a[i]]=true;
            add(i);
        }
    }
    for(int i=n;i;--i)
    {
        if(!have[a[i]]) nxt[i]=n+1;
        else
        {
            if(!last[a[i]]) 
            {
                nxt[i]=n+1;
                last[a[i]]=i;
            }
            else
            {
                nxt[i]=last[a[i]];
                last[a[i]]=i;
            }
        }
    }
    int m;
    read(m);
    for(int i=1;i<=m;++i) 
    {
        read(e[i].l);
        read(e[i].r);
        e[i].id=i;
    }
    sort(e+1,e+m+1,cmp);
    int j=1;
    for(int i=1;i<=m;++i)
    {
        while(j<e[i].l) add(nxt[j++]);
        ans[e[i].id]=query(e[i].r)-query(e[i].l-1);
    }
    for(int i=1;i<=m;++i) cout<<ans[i]<<'
';
}

1878: [SDOI2009]HH的项链

Time Limit: 4 Sec  Memory Limit: 64 MB
Submit: 5156  Solved: 2549
[Submit][Status][Discuss]

Description

HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运，所以每次散步 完后，他都会随意取出一

段贝壳，思考它们所表达的含义。HH不断地收集新的贝壳，因此他的项链变得越来越长。有一天，他突然提出了一

个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答。。。因为项链实在是太长了。于是，他只

好求助睿智的你，来解决这个问题。

Input

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

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

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

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

N ≤ 50000，M ≤ 200000。

Output

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

Sample Input

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

Sample Output

2
2
4