洛谷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 <= 500000，M <= 500000。

 

题解:

　　感觉没有见过的话很难想到。这个题目就是要我们求区间有多少种颜色，在一个区间里面，我们考虑对于多次出现的颜色，只统计最后出现的那一次，维护一个树状数组，每个位置用0或者1表示有没有贡献，因为对于多次出现的颜色，只统计最后出现的那一次。所以每次扫到一个元素位置为i，如果他之前没有出现过就直接给i这个位子贡献加1（insert(i,1)），如果之前出现过，就将上一次出现过的位置贡献减1，然后当前位置贡献加1。（add(las[a[i]],-1)，add(i,1)）。

 

代码：

　　

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <iostream>
#define MAXN 500010
#define ll long long
using namespace std;
ll tr[MAXN],las[1000010],a[MAXN],ans[MAXN];
int n,m;
struct xvwen{
    int l,r,id;
    void read(){
        scanf("%d%d",&l,&r);
    }
}q[MAXN];

int lb(int x){
    return x&-x;
}

void add(int now,int x){
    while(now<=n){
        tr[now]+=x;
        now+=lb(now);
    }
}

ll getsum(int now){
    ll ans=0;
    while(now){
        ans+=tr[now];
        now-=lb(now);
    }
    return ans;
}

bool cmp(xvwen x,xvwen y){
    if(x.r<y.r) return 1;
    return 0;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    scanf("%d",&m);
    for(int i=1;i<=m;i++) q[i].read(),q[i].id=i;
    sort(q+1,q+m+1,cmp);int now=1;
    for(int i=1;i<=n;i++){
        if(las[a[i]]) add(las[a[i]],-1);
        add(i,1);las[a[i]]=i;
        while(q[now].r==i){
            ans[q[now].id]=getsum(i)-getsum(q[now].l-1);
            now++;
        }
    }
    for(int i=1;i<=m;i++) printf("%lld\n",ans[i]);
    return 0;
}