2023牛客OI赛前集训营-提高组(第二场)B.出租

2023牛客OI赛前集训营-提高组(第二场)B.出租

B-出租_2023牛客OI赛前集训营-提高组(第二场) (nowcoder.com)

文章目录

  • 2023牛客OI赛前集训营-提高组(第二场)B.出租
    • 题目大意
    • 思路

题目大意

在一条路上有 n n n 个栋楼,每栋楼上有 k k k 个房间出租。

现在有 m m m 次询问,每次有两个数字 x , y x , y x,y ,询问区间 [ x , x + d ] [x , x +d] [x,x+d] 上是否有 y y y 个房间空闲。

思路

先思考一下 N O NO NO 的情况

如果存在 [ l , r ] [l , r] [l,r] 租户人数的和大于 k ∗ ( r − l + 1 + d ) k * (r - l + 1 + d) k(rl+1+d) ,说明无解

作差得到 ∑ l r ( v a l − k ) > k ∗ d \sum_l^r(val - k) > k *d lr(valk)>kd

所以我们只需要用线段树维护最大字段和和 k ∗ d k *d kd 比大小就好了

注意: 即便不能满足某组租户也要把它们记录下来。

#include 
#define LL long long
#define fu(x , y , z) for(int x = y ; x <= z ; x ++)
#define lp p << 1
#define rp p << 1 | 1
using namespace std;
const int N = 5e5 + 5;
const LL mod = 1e18 + 5;
int n , m , k , d;
struct Tr {
    LL ml , mr , mx , v;
} tr[N << 2];
void build (int p , int l , int r) {
    if (l == r) tr[p].v = -k;
    else {
        int mid = l + r >> 1;
        build (lp , l , mid);
        build (rp , mid + 1 , r);
        tr[p].v = tr[lp].v + tr[rp].v;
    }
}
void change (int p , int l , int r , int x , LL val) {
    if (l == r && l == x) {
        tr[p].v += val;
        tr[p].ml = tr[p].mr = tr[p].mx = tr[p].v;
    }
    else {
        int mid = l + r >> 1;
        if (x <= mid) change (lp , l , mid , x , val);
        else change (rp , mid + 1 , r , x , val);
        tr[p].v = tr[lp].v + tr[rp].v;
        tr[p].ml = max (tr[lp].ml , tr[rp].ml + tr[lp].v);
        tr[p].mr = max (tr[rp].mr , tr[lp].mr + tr[rp].v);
        tr[p].mx = max (tr[lp].mr + tr[rp].ml , max (tr[lp].mx , tr[rp].mx));
    }
}
int main () {
    int x;
    LL y;
    scanf ("%d%d%d%d" , &n , &m , &k , &d);
    build (1 , 1 , n);
    fu (i , 1 , m) {
        scanf ("%d%lld" , &x , &y);
        change (1 , 1 , n , x , y);
        if (tr[1].mx > k * d) puts ("NO");
        else puts ("YES");
    }
    return 0;
}

你可能感兴趣的:(线段树,题解,数据结构,数据结构,算法)