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]);
}
}