【洛谷P1972】HH的项链（求区间内有多少个不同的数---树状数组）

题目地址：https://www.luogu.org/problemnew/show/P1972

题目：

---

题目描述

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。

解题思路：

---

模版题，离线处理。

网上找到了两种做法，把我死磕了好久的那种方法当作第二种吧。

先对输入的【l,r】以r为关键字从小到大排序，一个虚拟数组cnt[]（之所以说它是虚拟数组，因为可以用树状数组实现想要的操作，不需要真的定义这么一个数组）。

--->第一种：

last[i]记录i位置上的数上一次出现的位置，当遍历给定序列的下标i时，cnt[i]+1,如果i位的这个数之前出现过，即存在last[i]，则将cnt[last[i]]-1（在遍历到i之前已经遍历过last[i]位置了cnt[last[i]]=1，所以这样操作下来，cnt[last[i]]=0),不会对当前要求的结果造成影响。（请参考代码理解，在区间右边界一定的前提下该解法是正确的）

比如序列：2，3，4，3，4，2

当右边界为5时（下标从1开始），对应的cnt[]值分别为：1，0，0，1，1

所以区间【1，5】内不同的数字个数=3，区间【2，5】内不同的数字个数=2，区间【4，5】内不同的数字个数=2（树状数组getsum求解）

因为右边界一定，所以右边界左侧不统计重复出现的数，每个数只统计一次，确保答案正确

--->第二种：

在第一种的基础上做了巧妙的变化，front[i]记录i位置上的数上一次出现的位置+1。。。有点不是很好理解。。目前我也没有完全理解透彻，暂且掌握第一种吧。

ac代码：

---

方法1：

#include <bits/stdc++.h>
using namespace std;
#define maxn 5000010
#define lowbit(x) ((x) & (-x))
int n, m, tmp;
int c[maxn],last[maxn],exist[maxn];
int ans[maxn];
struct node {
    int l,r,no;
    friend bool operator < (node a, node b)
    {
        return a.r == b.r ? a.l < b.l : a.r < b.r;
    }
}a[maxn];
void update(int x,int v)
{
    for(int i = x; i < maxn; i += lowbit(i))
        c[i] += v;
}
int query(int x)
{
    int sum=0;
    for(int i = x; i > 0; i -= lowbit(i))
        sum += c[i];
    return sum;
}
int main()
{
    //freopen("/Users/zhangkanqi/Desktop/11.txt", "r", stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&tmp);
        last[i] = exist[tmp];//记录上一个tmp出现的位置
        exist[tmp] = i;
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&a[i].l,&a[i].r);
        a[i].no=i;//每次查询的序号
    }
    sort(a+1,a+1+m);//按右端点排序
    int k=0;
    for(int i = 1; i <= m; i++)
    {
        while(k < a[i].r)
        {
            k++;
            if(last[k])
                update(last[k], -1);//把之前的值调0，1+（-1）
            update(k, 1);
        }
        ans[a[i].no] = query(a[i].r) - query(a[i].l - 1);
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}

方法2：

#include <bits/stdc++.h>
using namespace std;
#define maxn 5000010
#define lowbit(x) ((x) & (-x))
int n,m;
int c[maxn],front[maxn],last[maxn];
int ans[maxn];
struct node {
    int l,r,no;
    friend bool operator < (node a, node b)
    {
        return a.r == b.r ? a.l < b.l : a.r < b.r;
    }
}a[maxn];
void update(int x,int v)
{
    for(int i = x; i < maxn; i += lowbit(i))
        c[i] += v;
}
int query(int x)
{
    int sum=0;
    for(int i = x; i > 0; i -= lowbit(i))
        sum += c[i];
    return sum;
}
int main()
{
    //freopen("/Users/zhangkanqi/Desktop/11.txt", "r", stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int temp;
        scanf("%d",&temp);
        front[i]=last[temp]+1;//front数组为i位置上的数上一次出现的位置+1
        last[temp]=i;//last数组为i位置上的数到目前为止最后一次出现的位置
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&a[i].l,&a[i].r);
        a[i].no=i;//每次查询的序号
    }
    sort(a+1,a+1+m);//按右端点排序
    int k=1;//k初值为1
    for(int i=1;i<=m;i++)
    {
        while(k<=a[i].r)
        {
            k++;
            update(front[k-1],1);
            update(k,-1);
        }
        ans[a[i].no]=query(a[i].l);//查询左端点
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}

【参考博客】：

https://blog.csdn.net/xqx1343002589/article/details/81296741

https://blog.csdn.net/sunshiness_s/article/details/80429470