cf edu round 87 d (树状数组新模板)

19 篇文章 0 订阅
3 篇文章 0 订阅

我恨张宇昊,题读错了害得我们搞半天做不出来

拓展树状数组:支持O(lgn)单点修改;O(lgn)求前x个数的和;O(lgn)求前x个数的和等于res,已知res求x

题意:
告诉你n,q,再给你n个数,把这些数放入一个multiset(可重复的有序集合),给你q个数(ki),如果ki>0,那么把ki放入集合,如果ki<0,那么把集合里第-ki个数删去,问你最后集合有那些数,随便输出一个就可以了,没有就输出0
(1≤n,q≤106, 1≤a1≤a2≤⋯≤an≤n, ki≤n)

输入1
5 4
1 2 3 4 5
-5 -1 -3 -1
输出1
3

输入2
6 2
1 1 1 2 3 4
5 6
输出2
6

思路:
因为是有序的,而且范围很小,所以用计数排序的思想,搞个数组bit,bit[i]表示集合里面有几个i,插入的时候就bit[i]++,找第k个数的时候就通过前缀和,然后二分搜索,
相当于既需要单点修改,有需要前缀和,那么树状数组是首选,

这是搜索第k个数的代码复杂度是O(lgn*lgn)

int find(int k) {
	int l = 1, r = n, ans = n;
	while (l <= r) {
		int mid = l + r >> 1;
		if (sum(mid) >= k)ans = mid, r = mid - 1;
		else l = mid + 1;
	}
	return ans;
}

这样写是可以ac的,但是如果你这样写,说明你对树状数组和二进制的理解不够深刻

树状数组里bit[(1<<i)]的值等于前(1<<i)个节点的前缀和,而任意一个数k都是由无数个(1<<i)组成的,利用这个性质可以写出一下代码,要仔细解释的话要搞很久,懒得解释,直接上代码把

#include <iostream>

using namespace std;

int maxn = 1000001;
int n, tmp, bits[1000011], num, q;
void add(int x, int v) {
	for (int i = x; i <= maxn; i += (i&-i))
		bits[i] += v;
}
int sum(int x) {//前x个数的和等于res,已知x求res
	int res = 0;
	for (int i = x; i; i -= (i&-i))
		res += bits[i];
	return res;
}
int find(int res){//前x个数的和等于res,已知res求x
	int x = 0;
	for (int i = 19; i >= 0; --i){
		if (x + (1 << i) > maxn || bits[x + (1 << i)] > res - 1)//为何这里是res-1而不是res,因为你要找的是sum(x)=res且x最小
			continue;
		x += (1 << i);
		res -= bits[x];
	}
	return x + 1;//找不到的时候会返回maxn
}

int main(){
	scanf("%d%d", &n, &q);
	//maxn = n;
	for (int i = 1,a; i <= n; ++i)
		scanf("%d", &a), add(a, 1);
	
	for (int i = 1,k; i <= q; ++i){
		scanf("%d", &k);
		if (k > 0) add(k, 1);
		else add(find(-k), -1);
	}
	if (sum(n) == 0) { printf("0\n"); }
	else printf("%d\n", find(1));
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值