Codeforces Round #220 (Div. 2) D 树状数组 && 二分

/*题目*/

题意：给了n,m,然后一个包含m个数的数组nnum，数组默认从小到大排序，然后是 n个操作，输入一个数x，若x为0，把0加到这个字符串的末尾，若x为1，把1加到这个字符串的末尾，若x为-1，那么把字符串里的 下标 与 nnum数组里的元素相等的  给删除，字符串一开始是空的，问你最后字符串里有什么，若为空 就输出 POOR STACK

这题目看这操作一般都很容易联想到线段树，树状数组，一开始我建了个树状数组，但是超时了，毕竟操作很多，而且 在删除操作里，若nnum数组很大，等同于10^12的复杂度，只能优化，一般来说是用二分了，直接以这个串的某个位置是否为空来建立树状数组，每一次加在最后就不用说了，直接加就可以，关键在于删除，首先删除，要第一步二分查找出 要删除的最后一位，比如字符串长度为7，而nnum数组里 有个8，其实8这个位置不需要删除的，但是这个还是超时，于是想到在树状数组里删除的时候也需要二分，可是写了一直错，后来借鉴了杰哥的发现，原来WA在这里，比如 当前字符串为01010，需要删除 1，3，4位置的字符，要轮着来，当你删除1号位置的时候，字符串变成了1010，你在树状数组里也是要删除1节点的状态，所以原来的3号位置在当前变成了2号位置，当你删了3号以后变成了110 ，原来的4号位置变成了2号位置，所以每删除一次，下标会发生变化，但是没事，发现前面操作了几次而你原来的下标就比当前 减少了几，所以 nnum[i] - i 其实就是 当前需要删除的 元素 在 树状数组里的位置

int n,m;

int c[1000000 + 55];

int nnum[1000000 + 55];

int aa[1000000 + 55];

int sum[1000000 + 55];

int len;

void init() {
	memset(c,0,sizeof(c));
	memset(aa,-1,sizeof(aa));
}

bool input() {
	while(cin>>n>>m) {
		for(int i=0;i<m;i++)cin>>nnum[i];
		return false;
	}
	return true;
}

int lowbit(int x) {
	return x&(-x);
}

void add(int i,int val) {
	while(i <= n) {
		c[i] += val;
		i += lowbit(i);
	}
}

int get_sum(int i) {
	int sum = 0;
	while(i > 0) {
		sum += c[i];
		i -= lowbit(i);
	}
	return sum;
}

void binary_find(int pos,int id) {
	int l = 1;
	int r = n;
	while(l <= r) {
		int mid = (l + r)>>1;
		int ret = get_sum(mid);
		if(aa[mid] == -1) {
			if(ret >= nnum[pos] - id) r = mid - 1;
			else l = mid + 1;
		}
		else if(ret == nnum[pos] - id) {
			add(mid,-1);
			aa[mid] = -1;
			len--;
			break;
		}
		else if(ret > nnum[pos] - id) r = mid - 1;
		else if(ret < nnum[pos] - id) l = mid + 1;
	}
}

void cal() {
	int q = n;
	len = 0;
	int cnt = 1;
	while(q--) {
		int type;
		cin>>type;
		if(type == -1) {
			if(len == 0)continue;
			if(nnum[0] > len)continue;
			int now = lower_bound(nnum,nnum + m,len) - nnum;
			for(int i=0;i<=now;i++)
				binary_find(i,i);
		}
		else {
			aa[cnt] = type;
			add(cnt,1);
			cnt++;
			len++;
		}
	}
	if(len <= 0){puts("Poor stack!");return ;}
	for(int i=1;i<=n;i++)
		if(aa[i] != -1)
			printf("%d",aa[i]);
	puts("");
}

void output() {

}

int main() {
	while(true) {
		init();
		if(input())return 0;
		cal();
		output();
	}
	return 0;
}