codeforces 46D 线段树 区间合并

和poj 的hotel差不多，多了一些细节处理的过程

给一辆汽车安排空位的时候还需要和前后的车都保持一定的车距，如果前面没车或者后面没车，则可以停在边界上

典型的线段树区间合并

#include<cstdio>
#include<cstring>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn = 200010;
int sum[maxn<<2],col[maxn<<2];
int lsum[maxn<<2],rsum[maxn<<2];
map<int,pair<int,int> >re;
inline int max(int a,int b){
    return a>b?a:b;
}
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(sum[rt<<1],sum[rt<<1|1]);
    sum[rt]=max(rsum[rt<<1]+lsum[rt<<1|1],sum[rt]);
}
void pushdown(int rt,int m){
    if(col[rt]!=-1){
        lsum[rt<<1]=rsum[rt<<1]=sum[rt<<1]=col[rt] ? m-(m>>1) : 0 ;
        lsum[rt<<1|1]=rsum[rt<<1|1]=sum[rt<<1|1]=col[rt] ? (m>>1): 0;
        col[rt]=-1;
    }
}
void build(int l,int r,int rt){
    lsum[rt]=rsum[rt]=sum[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 c,int l,int r,int rt){
    if(L<=l&&r<=R){
        lsum[rt]=rsum[rt]=sum[rt]= c ? r-l+1:0;
        col[rt]=c;
        return ;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m) update(L,R,c,lson);
    if(R>m) update(L,R,c,rson);
    pushup(rt,r-l+1);
}
int query(int w,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]>=w) return query(w,lson);
    else if(rsum[rt<<1]+lsum[rt<<1|1]>=w) return m-rsum[rt<<1]+1;
    else return query(w,rson);
}
int main()
{
    int n,m,behind,front,op,a,b,pos,ops;
    while(scanf("%d%d%d",&n,&behind,&front)!=EOF)
    {
        ops=0;
        build(0,n-1+front,1);
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d%d",&op,&a);
            ops++;
            if(op==1)
            {
                if(sum[1]>=a+behind+front || lsum[1] >= a+front)
                {
                    if(lsum[1]>=a+front) pos=0;
                    else  pos=query(a+behind+front,0,n-1+front,1);
                    if(pos!=0) 
                    {
                        printf("%d\n",pos+behind);
                        update(pos+behind,pos+behind+a-1,0,0,n-1+front,1);
                        re[ops]=make_pair(pos+behind,pos+behind+a-1);
                    }
                    else
                    {
                        printf("0\n");
                        update(0,a-1,0,0,n-1+front,1);
                        re[ops]=make_pair(0,a-1);
                    }
                }
                else printf("-1\n");
            }
            else
            {
                int l=(int)re[a].first,r=(int)re[a].second;
                update(l,r,1,0,n-1+front,1);
            }
        }
    }
    return 0;
}

 

 

哥用线段树调了n久，神牛们却用如此简短的代码让我 汗颜

分析：因为需要插入的次数很小，所以可以直接记录，每辆汽车一次插入的位置，被占据的位置和空位间肯定是一段一段相邻的

即，比如有空位 a-b、c-d,       b-c  被占据了，如果a-b不够长，那么指针就要往右移动两个位置了，即跳过中间被占据的位置

#include <cstdio>
#include <cstring>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
int L, B, F;
int pos[105][2], tm=0;
set<int> s;
int main(void) {
    scanf("%d%d%d", &L, &B, &F);
    s.insert(-B);
    s.insert(L+F);
    int n;
    scanf("%d", &n);
    while(n--) {
        int x, y;
        scanf("%d%d", &x, &y);
        ++tm;
        if (x == 1) {
            int f=0;
            for(set<int>::iterator it=s.begin(); it!=s.end(); ) {
                set<int>::iterator nxt=it;
                ++nxt;
                if (nxt!=s.end() && ((*it)+B+y+F <= *nxt)) {
                    printf("%d\n", (*it)+B);
                    s.insert((*it)+B);
                    s.insert((*it)+B+y);
                    pos[tm][0] = *it+B;
                    pos[tm][1] = *it+B+y;
                    f=1;
                    break;
                }
                ++it; if(it!=s.end()) ++it;
            }
            if(!f) puts("-1");
        } else {
            s.erase(pos[y][0]);
            s.erase(pos[y][1]);
        }
    }
    return 0;
}