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

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

拓展树状数组:支持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≤10⁶, 1≤a₁≤a₂≤⋯≤a_n≤n, k_i≤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;
}