其他分享
首页 > 其他分享> > 酒店

酒店

作者:互联网

题目描述

A城市有个超级大酒店,住房部只有一层,只向南,但房间有N个(编号1, 2 .. N)。刚开始所有房间都是空的,但酒店肯定会有人来住宿,有些房间就要被入住。但客人经常会问一个问题,最长连续的空房间是多少,因为客人想连续的住在一起。于是,有下列三个操作:

1 a b:表示从房间a开始连续b个房间,将被入住(不管有没有人已经入住)

2 a b:表示从房间a开始连续b个房间,将清空入住(不管有没有人入住)

3:表示询问当前最长连续的空房间是多少?

对于3的询问,输出相应答案。

 

输入

首先输入N和P,  N如问题描述,P表示有多少个操作

然后输入P个操作

 

输出

对于第三种操作,输出相应答案

 

样例输入

12 10
3
1 2 3
1 9 4
3
2 2 1
3
2 9 2
3
2 3 2
3

样例输出

12
4
4
6
10

提示

 

【数据规模和约定】


3<=N<=16,000


3<=P<=200,000

 

 

首先,大家都能看出来,这是一道线段树的题。

那么除了常规的区记录间左右端点的数组,lazy数组,还要有一个数组sum[],表示所在区间内最长连续的空房间数。但是这并不够,在区间合并的时候很容易证明sum[now] = sum[now << 1] + sum[now << 1 | 1]是错的,这里不解释。

所以对于每一个区间,还要维护两个值:一个是从左端点开始最长连续空房间数lmax,另一个是从右端点开始最长连续空房间数rmax,举个栗子:

对于区间[1, 14]:

      1 2 3 4 5 6 7 8 9 10 11 12 13 14 

               # # #                          #            

'#' 为已经入住的个房间

此时sum[now] = 6, lmax = 3, rmax = 1.

那么对于维护sum[now],就可以得出

    sum[now] = max(sum[now << 1], sum[now << 1 | 1], rmax[now << 1] + lmax[now << 1 | 1])

也就是说要考虑连个子区间的空房间刚好连在一块的情况。

 

接下来就是想办法维护lmax[now]和rmax[now].

这个也不难(但我还是智障了debug了好长时间),想清楚了就行了。

拿lmax[now]举例,只要lmax[now << 1]是整个区间的长度的话,即now << 1整个区间都空,那么lmax[now] = lmax[now << 1] + lmax[now << 1 | 1],否则lmax[now] = lmax[now << 1].

 

rmax[now]同理。

 

最后想提一下lazy标记,和平时写的稍有些不同,不是累加的关系,而是将lazy[now]直接覆盖到lazy[now << 1]和lazy[now << 1 | 1]。因为对于一个房间只可能有入住和不入住两种情况,而这两种情况互不影响(题中描述如此:“不管有没有人已经入住”)。

 

所以更新的函数就写出来了。

更新,以及pushdown

 1 void pushdown(int now) 2 { 3     if(!a[now].lazy) return; 4     a[now << 1].lazy = a[now << 1 | 1].lazy = a[now].lazy; 5     a[now].lazy = 0; 6     if(a[now].lazy == 2)        //清空入住  7     { 8         a[now << 1].lmax = a[now << 1].rmax = a[now << 1].sum = a[now << 1].len; 9         a[now << 1 | 1].lmax = a[now << 1 | 1].rmax = a[now << 1 | 1].sum = a[now << 1 | 1].len;10     }11     else12     {13         a[now << 1].lmax = a[now << 1].rmax = a[now << 1].sum = 0;14         a[now << 1 | 1].lmax = a[now << 1 | 1].rmax = a[now << 1 | 1].sum = 0;15     }16 }17 void update(int L, int R, int now, int d)18 {19     if(L == a[now].l && R == a[now].r)20     {21         if(d == 2) a[now].lmax = a[now].rmax = a[now].sum = R - L + 1;22         else a[now].lmax = a[now].rmax = a[now].sum = 0;23         a[now].lazy = d; return;        //直接覆盖lazy 24     }25     pushdown(now);26     int mid = (a[now].r + a[now].l) >> 1;27     if(R <= mid) update(L, R, now << 1, d);28     else if(L > mid) update(L, R, now << 1 | 1, d);29     else {update(L, mid, now << 1, d); update(mid + 1, R, now << 1 | 1, d);}30     a[now] = a[now << 1] + a[now << 1 | 1];            //因为用了结构体,所以这里重载了一下加号。 31 }

啊对了,补一下定义的结构体和重载运算符

struct Tree
{    int l, r, lazy, len;        //len:区间长度 
    int sum, lmax, rmax;    /*其实这种封装也有一定的弊端,因为在重载加号的时候,返回的是一个新的结构体,
    所以必须对他的所有成员都赋上相应的值,否则会出错,具体看下方注释 */ 
    Tree operator + (const Tree& other)const
    {
        Tree ret;
        ret.sum = max(max(sum, other.sum), rmax + other.lmax);
        ret.l = l; ret.r = other.r;
        ret.len = ret.r - ret.l + 1;        //这就是容易疏忽的一点,我当时就忘了给len赋值,debug了1个多小时 
        if(lmax == len) ret.lmax = lmax + other.lmax;    //分情况维护lmax,rmax 
        else ret.lmax = lmax;        if(other.rmax == other.len) ret.rmax = other.rmax + rmax;        else ret.rmax = other.rmax;
        ret.lazy = 0;    //因为这条语句是在pushdown之后执行的,所以lazy已经被清空了 
        return ret;    
        /*如此看来,结构体里就应该有sum,lmax,rmax,因为其他都是不变的,或是在pushup操作里不需要的
        这么做也是尽量减少运算次数进行常数优化,但最重要的是,这样显得思维清晰,代码简练*/ 
    }
}a[maxn << 2];

 对于查询,只用返回a[1].sum了,因为是全局查询。

 

完整版代码

  1 #include  2 #include  3 #include  4 #include  5 #include  6 #include  7 using namespace std;  8 #define enter printf("\n")  9 #define space printf(" ") 10 typedef long long ll; 11 const int INF = 0x3f3f3f3f; 12 const int maxn = 2e4 + 5; 13 inline ll read() 14 { 15     ll ans = 0; 16     char ch = getchar(), last = ' '; 17     while(!isdigit(ch)) {last = ch; ch = getchar();} 18     while(isdigit(ch)) 19     { 20         ans = ans * 10 + ch - '0'; ch = getchar();    
 21     } 22     if(last == '-') ans = -ans; 23     return ans; 24 } 25 inline void write(ll x) 26 { 27     if(x < 0) x = -x, putchar('-'); 28     if(x >= 10) write(x / 10); 29     putchar('0' + x % 10); 30 } 31  32 int n, p; 33  34 struct Tree 35 { 36     int l, r, lazy, len;        
 37     int sum, lmax, rmax; 38     Tree operator + (const Tree& other)const 39     { 40         Tree ret; 41         ret.sum = max(max(sum, other.sum), rmax + other.lmax); 42         ret.l = l; ret.r = other.r; 43         ret.len = ret.r - ret.l + 1;        
 44         if(lmax == len) ret.lmax = lmax + other.lmax;    
 45         else ret.lmax = lmax; 46         if(other.rmax == other.len) ret.rmax = other.rmax + rmax; 47         else ret.rmax = other.rmax; 48         ret.lazy = 0;    
 49         return ret;    
 50     } 51 }a[maxn << 2]; 52 void build(int L, int R, int now) 53 { 54     a[now].l = L; a[now].r = R; 55     a[now].sum = a[now].lmax = a[now].rmax = a[now].len = R - L + 1; 56     a[now].lazy = 0; 57     if(L == R) return;    
 58     int mid = (L + R) >> 1; 59     build(L, mid, now << 1); 60     build(mid + 1, R, now << 1 | 1); 61 } 62 void pushdown(int now) 63 { 64     if(!a[now].lazy) return; 65     a[now << 1].lazy = a[now << 1 | 1].lazy = a[now].lazy; 66     a[now].lazy = 0; 67     if(a[now].lazy == 2)        
 68     { 69         a[now << 1].lmax = a[now << 1].rmax = a[now << 1].sum = a[now << 1].len; 70         a[now << 1 | 1].lmax = a[now << 1 | 1].rmax = a[now << 1 | 1].sum = a[now << 1 | 1].len; 71     } 72     else 73     { 74         a[now << 1].lmax = a[now << 1].rmax = a[now << 1].sum = 0; 75         a[now << 1 | 1].lmax = a[now << 1 | 1].rmax = a[now << 1 | 1].sum = 0; 76     } 77 } 78 void update(int L, int R, int now, int d) 79 { 80     if(L == a[now].l && R == a[now].r) 81     { 82         if(d == 2) a[now].lmax = a[now].rmax = a[now].sum = a[now].len; 83         else a[now].lmax = a[now].rmax = a[now].sum = 0; 84         a[now].lazy = d; return;        
 85     } 86     pushdown(now); 87     int mid = (a[now].r + a[now].l) >> 1; 88     if(R <= mid) update(L, R, now << 1, d); 89     else if(L > mid) update(L, R, now << 1 | 1, d); 90     else {update(L, mid, now << 1, d); update(mid + 1, R, now << 1 | 1, d);} 91     a[now] = a[now << 1] + a[now << 1 | 1];             
 92 } 93  94 int main() 95 { 96     n = read(); p = read(); 97     build(1, n, 1); 98     for(int i = 1; i <= p; ++i) 99     {100         int d = read();101         if(d == 3) {write(a[1].sum); enter;} 
102         else103         {104             int L = read(), add = read();105             update(L, L + add - 1, 1, d);106         }107     }108     return 0;109 }

 

 

标签:酒店,sum,rmax,ret,lmax,other,now
来源: https://blog.51cto.com/u_15234622/2830828