题意:有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;
}