【线段树】POJ 3667 Hotel 区间合并
水题好像来的晚了一些
题意:
1 a:询问是不是有连续长度为a的空房间,有的话住进最左边
2 a b:将[a,a+b-1]的房间清空
在最底层用1表示空,0表示占有这房间。。
sum[rt]表示该节点下最大连续空房间数
lsum[rt]表示该节点从左端开始连续的空房间数
rsum[rt]表示该节点从右端开始连续的空房间数
#include <stdio.h> #include <string.h> #include <stdlib.h> #include <math.h> #include <string> #include <iostream> #include <algorithm> using namespace std; #include <queue> #include <stack> #include <vector> #include <deque> #include <set> #include <map> #define IN freopen ("in.txt" , "r" , stdin); #define OUT freopen ("out.txt" , "w" , stdout); typedef long long LL; const int MAXN = 100110;//点数的最大值 const int MAXM = 20006;//边数的最大值 const int INF = 11521204; const int mod=1000000007; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 int sum[MAXN<<2],lsum[MAXN<<2],rsum[MAXN<<2]; int col[MAXN<<2]; void pushup(int rt,int m) { lsum[rt]=lsum[rt<<1]; rsum[rt]=rsum[rt<<1|1]; if(lsum[rt]==m-(m>>1)) lsum[rt]+=lsum[rt<<1|1]; if(rsum[rt]==m>>1) rsum[rt]+=rsum[rt<<1]; sum[rt]=max(max(sum[rt<<1],sum[rt<<1|1]),rsum[rt<<1]+lsum[rt<<1|1]); } void pushdown(int rt,int m) { if(col[rt]!=-1) { col[rt<<1]=col[rt<<1|1]=col[rt]; sum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=col[rt]?0:m-(m>>1); sum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=col[rt]?0:m>>1; col[rt]=-1; } } void build(int l,int r,int rt) { sum[rt]=lsum[rt]=rsum[rt]=r-l+1; col[rt]=-1; if(l==r) return ; int m=(l+r)>>1; build(lson); build(rson); } void update(int L,int R,int cc,int l,int r,int rt) { if(L<=l&&r<=R) { sum[rt]=lsum[rt]=rsum[rt]=cc? 0:r-l+1; col[rt]=cc; return ; } pushdown(rt,r-l+1); int m=(l+r)>>1; if(L<=m) update(L,R,cc,lson); if(m<R) update(L,R,cc,rson); pushup(rt,r-l+1); } int query(int x,int l,int r,int rt) { if(l==r) return l; pushdown(rt,r-l+1); int m=(l+r)>>1; if(sum[rt<<1]>=x) return query(x,lson); else if(rsum[rt<<1]+lsum[rt<<1|1]>=x) return m-rsum[rt<<1]+1; else return query(x,rson); } int main() { int n,m; //IN; while(scanf("%d%d",&n,&m)!=EOF) { build(1,n,1); while(m--) { int op,k,d; scanf("%d",&op); if(op==1) { scanf("%d",&d); if(sum[1]<d) printf("0\n"); else { int a=query(d,1,n,1); printf("%d\n",a); update(a,a+d-1,1,1,n,1); } } else { scanf("%d%d",&k,&d); update(k,k+d-1,0,1,n,1); } } } return 0; }