题目大意: 给n个单元内存,有 四个操作:
1、New x 找一段长为x的空区间段填满。 (下面用0表示空单元,1表示非空单元。)
2、Free x 释放包含x的区间段
3、 Get x 找到第x个区间段
4、将整个区间都置为空。
解题思路:可以说这题应该算是线段树之区间操作里面比较好的经典题了。要注意的地方很多。对vector容器的用法又了解了很多。
#include <cstdio>
#include <vector>
#include <iostream>
using namespace std;
#define lz 2*u,l,mid
#define rz 2*u+1,mid+1,r
const int maxn=50005;
int flag[4*maxn];
struct segment
{
int lm; // 从区间左端点开始的连续0个数;
int rm; // 以区间右端点结束的连续0个数;
int sm; // 整个区间最长连续的0个数;
} tree[4*maxn];
struct node
{
int s, d;
};
node tmp;
vector<node>vt;
void push_up(int u, int l, int r)
{
int mid= (l+r)>>1;
tree[u].lm= tree[u*2].lm;
tree[u].rm= tree[u*2+1].rm;
tree[u].sm= max( tree[u*2].sm, tree[u*2+1].sm );
if( tree[u*2].lm == mid-l+1 ) tree[u].lm += tree[u*2+1].lm;
if( tree[u*2+1].rm == r-mid ) tree[u].rm += tree[u*2].rm;
int t= tree[u*2].rm + tree[u*2+1].lm;
if( t > tree[u].sm ) tree[u].sm= t;
}
void push_down(int u, int l, int r)
{
if(flag[u]>=0)
{
int mid=(l+r)>>1;
flag[2*u]=flag[2*u+1]=flag[u];
tree[2*u].lm= tree[2*u].rm= tree[2*u].sm= flag[u]?0:mid-l+1;
tree[2*u+1].lm= tree[2*u+1].rm= tree[2*u+1].sm= flag[u]?0:r-mid;
flag[u]=-1;
}
}
void build(int u, int l, int r)
{
flag[u]=-1;
if(l==r)
{
tree[u].lm= tree[u].rm= tree[u].sm= 1;
return;
}
int mid= (l+r)>>1;
build(lz);
build(rz);
push_up(u, l, r);
}
void Update(int u, int l, int r, int tl, int tr, int c)
{
if(tl<=l&&r<=tr)
{
tree[u].lm=tree[u].rm=tree[u].sm=c?0:r-l+1;
flag[u]=c;
return ;
}
push_down(u,l,r);
int mid=(l+r)>>1;
if(tr<=mid) Update(lz,tl,tr,c);
else if(tl>mid) Update(rz,tl,tr,c);
else
{
Update(lz,tl,mid,c);
Update(rz,mid+1,tr,c);
}
push_up(u,l,r);
}
int Query(int u, int l, int r, int p)
{
if(tree[u].sm==p&&r-l+1==p) ///不仅要tree[u].sm==p而且要整个区间都被0覆盖,否则会出错
{
return l;
}
push_down(u,l,r);
int mid= (l+r)>>1, t;
if(p<=tree[2*u].sm) return Query(lz,p);
else if(tree[2*u].rm+tree[2*u+1].lm>=p) return mid-tree[2*u].rm+1;
else return Query(rz,p);
}
int find(int tp)
{
int l=0, r=vt.size()-1, mid, ans=-1;
while(l<=r)
{
int mid=(l+r)>>1;
if(vt[mid].s<=tp)
{
ans=mid;
l=mid+1;
}
else r=mid-1;
}
return ans;
}
int main()
{
int n, m;
while(cin >> n >> m)
{
build(1,1,n);
vt.clear();
while(m--)
{
char ch[6];
scanf("%s",ch);
if(ch[0]=='R')
{
vt.clear();
Update(1,1,n,1,n,0);
puts("Reset Now");
continue;
}
int op;
scanf("%d",&op);
if(ch[0]=='N')
{
if(tree[1].sm>=op)
{
int st=Query(1,1,n,op);
tmp.s=st, tmp.d=st+op-1;
int id=find(tmp.s);
vt.insert(vt.begin()+id+1,tmp);
printf("New at %d\n",tmp.s);
Update(1,1,n,tmp.s,tmp.d,1); ///!!!这里注意了,查询完了还要进行相应的更新
}
else
puts("Reject New");
}
else if(ch[0]=='G')
{
if(vt.size()>=op)
printf("Get at %d\n",vt[op-1].s);
else puts("Reject Get");
}
else if(ch[0]=='F')
{
int id=find(op);
if(id==-1||vt[id].d<op) puts("Reject Free");
else
{
Update(1,1,n,vt[id].s,vt[id].d,0);
printf("Free from %d to %d\n",vt[id].s,vt[id].d);
vt.erase(vt.begin()+id,vt.begin()+id+1); ///删除区间[s,d)内的元素,左闭右开
}
}
}
puts("");
}
return 0;
}