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 
#include 
#include 
#include 
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 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;
}


你可能感兴趣的:(数据结构)