0
点赞
收藏
分享

微信扫一扫

uvalive 5902(BIT)


题意:有n本书叠放,编号从上到下是1到n,每次操作是先输出编号是i这本书的上面有几本书,然后把这本书抽出来放到顶上。
题解:位置数组pos[i]记录编号i的书的位置,树状数组Sun(pos[i])统计位置i以下有多少本书,因为树状数组更新节点是往右走,所有对于节点i,右边节点应该表示节点i上面的书,所以位置数组要逆序,每次把书取出来前先把右边的节点更新,然后当前的书要放到最后一个节点的后面,后面节点还原。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 200005;
int n, m, C[N], pos[N];

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

int Sum(int x) {
    int ret = 0;
    while (x > 0) {
        ret += C[x];
        x -= lowbit(x);
    }
    return ret;
}

void Add(int x, int d) {
    while (x <= N) {
        C[x] += d;
        x += lowbit(x);
    }
}

int main() {
    int t;
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &m);
        memset(C, 0, sizeof(C));
        for (int i = 1, j = n; i <= n; i++, j--) {
            pos[i] = j;
            Add(i, 1);
        }
        int x, top = n;
        for (int i = 0; i < m; i++) {
            scanf("%d", &x);
            printf("%d", n - Sum(pos[x]));
            if (i != m - 1)
                printf(" ");
            Add(pos[x], -1);
            pos[x] = ++top;
            Add(top, 1);
        }
        printf("\n");
    }
    return 0;
}


举报

相关推荐

0 条评论