codeforces 374D 树状数组或者线段树

题意：给你n个操作，1，0，分别为在字符串上加上1，0。-1为删除再位置上ai的字符。

分析：单点更新。考虑树状数组，不过这个题目和一般的不同，需要二分，而线段树本身在更新和查询操作的时候就是相当于二分。所以线段树更容易理解。

具体详见代码：

线段树

#include<bits/stdc++.h> 
#define lson l ,mid ,rt<<1
#define rson mid+1,r,rt<<1|1
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 20;
int n,A[maxn<<2],B[maxn<<2],s;
struct IntexTree
{
	int sum[maxn<<2];
	void maintain(int rt)
	{
		sum[rt] = sum[rt<<1] + sum[rt<<1|1];
	}
	void update(int q,int v,int l,int r,int rt)
	{
		if(l == r)
		{
			sum[rt] = v;
			B[l] = s;
			return ;
		}
		int mid = (l + r) >>1;
		if(q <= mid)	update(q,v,lson);
		else	update(q,v,rson);
		maintain(rt);
	}
	void remove(int q,int v,int l,int r,int rt)
	{
		if(l == r)
		{
			sum[rt] = v;
			return ;
		}
		int mid = (l + r) >>1;
		if(q <= sum[rt<<1])	remove(q,v,lson);
		else	remove(q -sum[rt<<1] ,v,rson);
		maintain(rt);
	}
	void query(int l,int r,int rt)
	{
		if(l == r && sum[rt])
		{
			printf("%d",B[l]);
			return ;
		}
		int mid = (l + r) >>1;
		if(sum[rt<<1])	query(lson);
		if(sum[rt<<1|1])	query(rson);
	} 
}tree;

int main(void)
{
	int m;
	cin>>n>>m;
	for(int i = 0;i<m;i++)
		scanf("%d",A+i);
	int q,k = 0;
	for(int  j = 0;j<n;j++)
	{
		scanf("%d",&q);
		s = q;
		 if(q == -1)	
		 {
		 	for(int i = 0;i<m;i++)
		 	{
		 		if(tree.sum[1] < A[i] - i)	break;
		 			tree.remove(A[i] -i,0,1,n,1);
		 	}
		 }
		 else if(q == 1) 
		 {
		 	tree.update(++k,1,1,n,1);
		 }
		 else	if(q == 0)
		 {
		 	tree.update(++k,1,1,n,1);
		 } 
	}
	if(tree.sum[1] == 0)	cout<<"Poor stack!";
	else
	tree.query(1,n,1);
} 

树状数组

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 20;
int n,m,A[maxn],B[maxn];
struct BIT
{
	int bit[maxn];
	void add(int x,int v)
	{
		while(x <= n)
		{
			bit[x] += v;
			x += x&-x;
		}
	}
	int sum(int x)
	{
		int ans = 0;
		while(x)
		{
			ans += bit[x];
			x -= x&-x;
		}
		return ans;
	}
	int remove(int x,int pos)
	{
		int L = 1,R = pos,mid = 0;
		while(R > L)
		{
			mid = (L + R) >>1;
			if(sum(mid) < x)  L = mid + 1;
			else R = mid;
		}
		add(L,-1);
		return L;
	}
}tree;
int main(void)
{
	cin>>n>>m;
	for(int i = 0;i<m;i++)
		scanf("%d",&A[i]);
	int q,k=0;
	for(int i = 0;i<n;i++)
	{
		scanf("%d",&q);
		if(q == -1)
		{
			int x = tree.sum(k);
			for(int i = 0;i<m;i++)
			{
				if(A[i] > x)	break;
				B[tree.remove(A[i] - i,k)] = -1;
			}
		}
		else
		{
			tree.add(++k,1);
			B[k] = q;
		}
	}
	int x = tree.sum(k);
	if(x ==0)	cout<<"Poor stack!" ;
	else
	for(int i = 1;i<=k;i++)
	{
		if(~B[i])	printf("%d",B[i]); 
	}
}