bzoj1878 HH的项链 树状数组

大致题意：给出一个序列num和一系列询问，每个询问是一个区间，要求输出此区间的不同的数的个数。

先把题目换一下，假设每个询问左端点固定是1，那么我们只需要用数组c记录一下每一个数字是否是第一次出现，如果第i个数是第一次出现，那么c[i]置为1，否则c[i]置为0，然后对于一个询问[1, x]，我们只需要计算c的前缀和即可。

然后回到本题，当左端点变化的时候，c数组出现了问题，此时的"第一次出现"应该从现在变化后的左端点开始计算，所以，再开一个数组next，next[i] = j，表示num[i] = num[j]且区间[i + 1, j - 1]这个区间中没有出现过num[i]，那么只要通过next来更改数组c即可，具体看代码。

#include <cstdio>
#include <map>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 50010;
const int maxm = 200010;
int N, M;
int c[maxn];
int num[maxn], next[maxn], ans[maxm];
struct node
{
    int x, y, id;
}Q[maxm];

int cmp(const node& A, const node& B)
{
    return A.x < B.x;
}

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

void add(int x, int v)
{
    while(x <= N)
    {   
        c[x] += v;
        x += lowbit(x);
    }
}

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

int main()
{
    map<int, int> hash;
    scanf("%d", &N);
    for (int i = 1; i <= N; i++)
        scanf("%d", &num[i]);
    scanf("%d", &M);
    for (int i = 0; i < M; i++)
    {
        scanf("%d%d", &Q[i].x, &Q[i].y);
        Q[i].id = i;
    }
    sort(Q, Q + M, cmp);
    hash.clear();
    for (int i = 1; i <= N; i++)
    {
        if (!hash.count(num[i]))
        {
            hash[num[i]] = i;
            add(i, 1);
        }
    }
    hash.clear();
    for (int i = N; i > 0; i--)
    {
        if (!hash.count(num[i])) next[i] = N + 1;
        else next[i] = hash[num[i]];
        hash[num[i]] = i;
    }

    int t = 1;
    for (int i = 0; i < M; i++) 
    {
        for (; t < Q[i].x; t++)
            add(next[t], 1);
        ans[Q[i].id] = sum(Q[i].y) - sum(Q[i].x - 1);
    }
    for (int i = 0; i < M; i++)
        printf("%d\n", ans[i]);
    return 0;
}