线段树 区间合并 poj 3667 Hotel

标签：

题意：初始时n个空房间，有m次操作，分为两种：1.找出最靠左的连续d个空房间的左端点值，并把该区间的房间占用  2.使指定区间的房间不被占用。输出每次操作1的左端点值，如果找不到，输出0。

很明显的一个思路是把每个节点所在区间的最长的连续空房间和该区间是否被覆盖作为节点信息

对于操作1，如果当前节点的值小于d，那当前区间肯定不行。如果不小于d,先看左子节点的值是否大于等于d（因为要找出最靠左的连续区间），如果大于等于，就到左子树中去找，但小于的时候就出现了麻烦。因为小于的时候不一定要找的区间就在右子树中，还有可能是左子树的右边一部分和右子树的左边一部分连起来形成的连续区间。所以我们应该再设两个数组lsum和rsum，分别保存节点所在区间从左边开始的连续空房间数和右边开始的连续房间数，这样如果左节点的rsum和右节点的lsum加起来大于等于d，就说明该连续区间就是要找的

对于操作2，就是普通的区间更新操作

特别应该注意的是pushdown和pushup操作：

pushdown:当把节点信息向下传递的时候，需要传递我们之前提到的四个信息，如果当前节点被覆盖了，那向下传递的时候左右子节点都应该为被覆盖，它们的连续空房间都应为0，lsum和rsum也应为0.如果当前节点没被覆盖，那向下传递的时候左右子节点也应该都没覆盖，连续空房间数和lsum和rsum都为它们所在的区间长度。

pushup:向上传递的时候，如果左子结点的lsum值小于左子结点的长度，那么父节点的lsum就等于左子结点的lsum，否则等于左子结点的lsum+右子节点的lsum。如果右子节点的rsum小于右子节点的区间长度，那父节点的rsum就等于右子节点的rsum，否则等于右子节点的rsum+左子结点的rsum。

易错点是注意每次更新信息要四个信息都更新，还有就是如果整个区间的最长连续空房间都小于d，自然输出0，这时候不应该再进行查询和更新操作。

代码：

#include <cstdlib>
#include <cctype>
#include <cstring>
#include <cstdio>
#include <cmath>
#include<climits>
#include <algorithm>
#include <vector>
#include <string>
#include <iostream>
#include <sstream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <fstream>
#include <numeric>
#include <iomanip>
#include <bitset>
#include <list>
#include <stdexcept>
#include <functional>
#include <utility>
#include <ctime>
using namespace std;

#define PB push_back
#define MP make_pair

#define REP(i,x,n) for(int i=x;i<(n);++i)
#define FOR(i,l,h) for(int i=(l);i<=(h);++i)
#define FORD(i,h,l) for(int i=(h);i>=(l);--i)
#define SZ(X) ((int)(X).size())
#define ALL(X) (X).begin(), (X).end()
#define RI(X) scanf("%d", &(X))
#define RII(X, Y) scanf("%d%d", &(X), &(Y))
#define RIII(X, Y, Z) scanf("%d%d%d", &(X), &(Y), &(Z))
#define DRI(X) int (X); scanf("%d", &X)
#define DRII(X, Y) int X, Y; scanf("%d%d", &X, &Y)
#define DRIII(X, Y, Z) int X, Y, Z; scanf("%d%d%d", &X, &Y, &Z)
#define OI(X) printf("%d",X);
#define RS(X) scanf("%s", (X))
#define MS0(X) memset((X), 0, sizeof((X)))
#define MS1(X) memset((X), -1, sizeof((X)))
#define LEN(X) strlen(X)
#define F first
#define S second
#define Swap(a, b) (a ^= b, b ^= a, a ^= b)
#define Dpoint  strcut node{int x,y}
#define cmpd int cmp(const int &a,const int &b){return a>b;}

 /*#ifdef HOME
    freopen("in.txt","r",stdin);
    #endif*/
const int MOD = 1e9+7;
typedef vector<int> VI;
typedef vector<string> VS;
typedef vector<double> VD;
typedef long long LL;
typedef pair<int,int> PII;
//#define HOME

int Scan()
{
	int res = 0, ch, flag = 0;

	if((ch = getchar()) == '-')				//判断正负
		flag = 1;

	else if(ch >= '0' && ch <= '9')			//得到完整的数
		res = ch - '0';
	while((ch = getchar()) >= '0' && ch <= '9' )
		res = res * 10 + ch - '0';

	return flag ? -res : res;
}
/*----------------PLEASE-----DO-----NOT-----HACK-----ME--------------------*/
#define maxn 50010
int msum[maxn<<2],lsum[maxn<<2],rsum[maxn<<2];
int cover[maxn<<2];
void pushup(int rt,int m)
{lsum[rt]=lsum[rt<<1];
rsum[rt]=rsum[rt<<1|1];
if(lsum[rt<<1]==(m-(m>>1)))
    lsum[rt]+=lsum[rt<<1|1];
if(rsum[rt<<1|1]==(m>>1))
    rsum[rt]+=rsum[rt<<1];
msum[rt]=max(lsum[rt<<1|1]+rsum[rt<<1],max(msum[rt<<1],msum[rt<<1|1]));

}
void pushdown(int rt,int m)
{
    if(cover[rt]!=-1)
    {
        cover[rt<<1]=cover[rt<<1|1]=cover[rt];
        msum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=cover[rt]?0:m-(m>>1);
        msum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=cover[rt]?0:m>>1;
        cover[rt]=-1;

    }
}
void build(int l,int r,int rt)
{
    cover[rt]=-1;
    msum[rt]=lsum[rt]=rsum[rt]=r-l+1;
    if(l==r)
        return;
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
}
void update(int l,int r,int rt,int a,int b,int c)
{
    if(a<=l&&r<=b)
    {
        msum[rt]=lsum[rt]=rsum[rt]=c?0:r-l+1;
        cover[rt]=c;
        return;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(a<=m)
        update(l,m,rt<<1,a,b,c);
    if(b>m)
        update(m+1,r,rt<<1|1,a,b,c);
    pushup(rt,r-l+1);
}
int  query(int l,int r,int rt,int d)
{int m=(l+r)>>1;
if(l==r)
    return l;
pushdown(rt,r-l+1);
if(msum[rt<<1]>=d)
return query(l,m,rt<<1,d);
else
    if(rsum[rt<<1]+lsum[rt<<1|1]>=d)
    return m-rsum[rt<<1]+1;
return query(m+1,r,rt<<1|1,d);

}


int main()
{int n,m;
RII(n,m);
build(1,n,1);
while(m--)
{
    int op;
    RI(op);
    if(op==1)
    {
        int d;
        RI(d);
        if(msum[1]<d)
            printf("0\n");
        else
        {int p=query(1,n,1,d);
        update(1,n,1,p,p+d-1,1);
        printf("%d\n",p);}
    }
    else
    {
        int x,d;
        RII(x,d);
        update(1,n,1,x,x+d-1,0);
    }
}



        return 0;
}

线段树 区间合并 poj 3667 Hotel

标签：

原文地址：http://blog.csdn.net/u013840081/article/details/46356015