UVa 1513 Movie collection 树状数组

题意:

\(n\)个影碟,每次操作先输出编号为\(x\)的影碟上面有多少影碟,然后把它抽出来放在最上面。

分析:

设编号为\(x\)的影碟的位置为\(pos_x\),每次查询前\(pos_x\)的前缀和,这是影碟\(x\)和下面影碟的数量,再用\(n\)减去就是答案。
然后把\(pos_x\)的值置为\(0\),再在最后面新加一个位置,置为\(1\)

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

const int maxn = 100000 + 10;

int n, m;
int C[maxn << 1];

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

void add(int x, int v) {
    while(x <= n + m) {
        C[x] += v;
        x += lowbit(x);
    }
}

int query(int x) {
    int ans = 0;
    while(x) {
        ans += C[x];
        x -= lowbit(x);
    }
    return ans;
}

int pos[maxn];

int main()
{
    int T; scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &m);
        memset(C, 0, sizeof(C));
        for(int i = 1; i <= n; i++) {
            pos[i] = n - i + 1;
            add(i, 1);
        }

        int tot = n;
        for(int i = 0; i < m; i++) {
            int x;
            scanf("%d", &x);
            printf("%d", n - query(pos[x]));
            add(pos[x], -1);
            add(++tot, 1);
            pos[x] = tot;
            if(i < m - 1) printf(" ");
            else printf("\n");
        }
    }

    return 0;
}
posted @ 2016-07-19 18:31  AOQNRMGYXLMV  阅读(...)  评论(... 编辑 收藏