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

说明/提示

对于所有数据，n,m<=1e6

分析:

主席树在线解法：点此

保存所有询问，并将询问按r排序。

每次插入数字的下标
每次查询保留上一次最后的位置last=r+1，并记录每个数字上一次出现的下标
对于每个询问，last-r之间的数字出现过，则删去上一次的，重新加上这一次的
这样就能保证数字在区间内只出现过一次

对于每个查询的R，我们只关注距离R近的数字，如果有两个相同的数字，则删掉距离R远的（显然距离R远的在之后的查询中相较于另外一个不利于生存）

详见代码

树状数组维护从1到x不同数字的个数
每个询问的答案就是ask(r )-ask(l-1)

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define lowbit(x) (x&(-(x)))
using namespace std;
const int maxm=1e6+6;
struct Node{
    int l,r;
    int id;
    bool operator <(const Node a){
        return r<a.r;
    }
}q[maxm];
int a[maxm];
int mark[maxm];//mark[i]表示i上一次出现的位置
int c[maxm];
int res[maxm];//保存结果
int n,m;
void add(int i,int t){
    while(i<=n){
        c[i]+=t;
        i+=lowbit(i);
    }
}
int ask(int i){
    int ans=0;
    while(i){
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
int main(){
    ios::sync_with_stdio(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    cin>>m;
    for(int i=1;i<=m;i++){
        cin>>q[i].l>>q[i].r;
        q[i].id=i;
    }
    sort(q+1,q+1+m);
    int last=1;//上次查询的结尾+1
    for(int i=1;i<=m;i++){
        for(int j=last;j<=q[i].r;j++){
            if(mark[a[j]]){
                add(mark[a[j]],-1);
            }
            add(j,1);
            mark[a[j]]=j;
        }
        last=q[i].r+1;
        res[q[i].id]=ask(q[i].r)-ask(q[i].l-1);
    }
    for(int i=1;i<=m;i++){
        cout<<res[i]<<endl;
    }
    return 0;
}