LA5902

好吧,用树状数组来做这个实在是太聪明了。好美妙的想法。

相当于统计区间和,假设有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;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值