好吧,用树状数组来做这个实在是太聪明了。好美妙的想法。
相当于统计区间和,假设有20W个书阁,如果第i个书阁有书的话就a[i] = 1,否则的话a[i] = 0;
首先想到的是按顺序放书,第一个书阁放第一部电影,第二个书阁放第二个电影,依次类推,那么统计第i个书阁前面的电影书只要统计从最前面到当前位置1的个数即可。
每看一部电影相当于把在首个书阁之前添加一个书阁。可以先把所以的电影都放在后100000个位置,第一部电影放第100000个书阁,第二部电影放第100001个书阁
如果看一部电影假设x,就统计x位置之前1的个数,然后用一个top来记录当前首部,看电影x就相当于把top位置置1,把当前电影x位置置0,然后top--;
初始化为当前于a[1000001],a[1000002]...为1,然后每次求和更新即可。
当然我们也可以倒过来做,发现倒过来做更加方便。
把最后一部电影把在第一个书阁的位置,倒数第二部摆在第二个书阁....
那么c[x] = lowbit(x)即可。就相当于初始化a[1],a[2]...为1了。
那么每次看一部电影就相当于a[top] = 1,top++;统计一下原始位置到看的电影位置的电影数目,相当于放在当前电影之后的数目,放在之前的数目只要用总数减去之后的就得到了。
AC代码:
#include <cstdio>
#include <string.h>
const int MAX_NUMBER = 200005;
int tree_array[MAX_NUMBER + 3], location[MAX_NUMBER];
int moive_number, watch_number;
int lowbit(int x) {
return x & (-x);
}
void init() {
for (int i = 1; i <= moive_number; i++) {
location[i] = moive_number - i + 1;
}
for (int i = 1; i <= MAX_NUMBER; i++) {
tree_array[i] = lowbit(i);
}
}
int getSum(int x) {
int ans = 0;
while (x > 0) {
ans += tree_array[x];
x -= lowbit(x);
}
return ans;
}
void update(int x, int d) {
while (x <= MAX_NUMBER) {
tree_array[x] += d;
x += lowbit(x);
}
}
int main() {
int test_case;
scanf("%d", &test_case);
while (test_case--) {
scanf("%d%d", &moive_number, &watch_number);
init();
int top = moive_number;
for (int i = 1; i <= watch_number; i++) {
int watch_moive;
scanf("%d", &watch_moive);
int cnt_location = location[watch_moive];
int ans = getSum(cnt_location);
if (i == 1) {
printf("%d", moive_number - ans);
}
else {
printf(" %d", moive_number - ans);
}
update(cnt_location, -1);
top++;
location[watch_moive] = top;
}
printf("\n");
}
return 0;
}