UESTC 1227 & POJ 3667 Hotel
非常细腻的线段树题目啊，后来还是有个细节写错了，查了一个晚上。。就不分析了。
代码：
#include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #include <utility> #include <cstdlib> using namespace std; #define N 80011 struct node { int ls,rs,ms; int pos; int mark; // 0: unsure 1: all-empty 2: all-full }tree[4*N]; int n,m; void build(int l,int r,int rt) { tree[rt].ls = tree[rt].rs = tree[rt].ms = r-l+1; tree[rt].pos = l; if(l == r) { return; } int mid = (l+r)/2; build(l,mid,2*rt); build(mid+1,r,2*rt+1); } int if_all_empty(int l,int r,int rt) { if(tree[rt].ls == r-l+1) return 1; return 0; } void update(int l,int r,int rt) { if(!tree[rt].mark) return; if(tree[rt].mark == 1) //全空，则下传给左右子树 { int len = r-l+1; tree[2*rt].ls = tree[2*rt].rs = tree[2*rt].ms = (len+1)/2; tree[2*rt].pos = l; tree[2*rt+1].ls = tree[2*rt+1].rs = tree[2*rt+1].ms = len/2; tree[2*rt+1].pos = (l+r)/2+1; tree[2*rt].mark = tree[2*rt+1].mark = 1; } else if(tree[rt].mark == 2) //全满，则下传给左右子树 { tree[2*rt].ls = tree[2*rt].rs = tree[2*rt].ms = 0; tree[2*rt].pos = l; tree[2*rt+1].ls = tree[2*rt+1].rs = tree[2*rt+1].ms = 0; tree[2*rt+1].pos = (l+r)/2+1; tree[2*rt].mark = tree[2*rt+1].mark = 2; } tree[rt].mark = 0; // not "== 0" } int query(int l,int r,int dis,int rt) { update(l,r,rt); if(tree[rt].ms<dis) return 0; if(tree[rt].ms == dis) return tree[rt].pos; int mid = (l+r)/2; if(tree[2*rt].ms>=dis) return query(l,mid,dis,2*rt); if(tree[2*rt].rs + tree[2*rt+1].ls>=dis) return mid - tree[2*rt].rs + 1; return query(mid+1,r,dis,2*rt+1); } void in_out(int l,int r,int aa,int bb,int flag,int rt) //flag == 1: insert else quit { if(aa>r||bb<l) return; if(aa<=l&&bb>=r) { if(flag == 1) //如果当前要入住 { tree[rt].ls = tree[rt].rs = tree[rt].ms = 0; tree[rt].pos = l; tree[rt].mark = 2; } else //如果当前要退房 { tree[rt].ls = tree[rt].rs = tree[rt].ms = r-l+1; tree[rt].pos = l; tree[rt].mark = 1; } return; } update(l,r,rt); int mid = (l+r)/2; in_out(l,mid,aa,bb,flag,2*rt); in_out(mid+1,r,aa,bb,flag,2*rt+1); tree[rt].ls = tree[2*rt].ls; if(if_all_empty(l,mid,2*rt)) tree[rt].ls += tree[2*rt+1].ls; tree[rt].rs = tree[2*rt+1].rs; if(if_all_empty(mid+1,r,2*rt+1)) tree[rt].rs += tree[2*rt].rs; tree[rt].ms = max(tree[2*rt].rs+tree[2*rt+1].ls,max(tree[2*rt].ms,tree[2*rt+1].ms)); if(tree[rt].ms == tree[2*rt].ms) //如果当前区间最大空房数等于左子树最大空房数 tree[rt].pos = tree[2*rt].pos; //则起点置为左子树的起点 else if(tree[rt].ms == tree[2*rt].rs + tree[2*rt+1].ls) //同理 tree[rt].pos = mid - tree[2*rt].rs + 1; else tree[rt].pos = tree[2*rt+1].pos; } int main() { scanf("%d%d",&n,&m); memset(tree,0,sizeof(tree)); build(1,n,1); int i,flag; int x,dis; for(i=0;i<m;i++) { scanf("%d",&flag); if(flag == 1) { scanf("%d",&dis); int ans = query(1,n,dis,1); printf("%d\n",ans); if(ans) in_out(1,n,ans,ans+dis-1,1,1); } else if(flag == 2) { scanf("%d%d",&x,&dis); in_out(1,n,x,x+dis-1,2,1); } } return 0; }