2021牛客暑期多校训练营6 H Hopping Rabbit 线段树，扫描线，矩形取模

代码

给你$n$个矩形，小兔子每次可以上下左右四个方向每步跳$d$个单位长度，问你是否能找到一初始点，使得小兔子无论如何跳都无法跳入陷阱。

分析

将每个矩形进行取模，实现对$[0,d-1]$的矩形进行覆盖。若在取模后的矩形存在未被覆盖的点则当前点可以作为初始点。

取模时注意，拆分后的矩形最多有四个，对$x1,x2$的值大小分情况讨论，$y1,y2$同理。

采用扫描线的方式进行扫描，线段树支持区间加和最小值。如果当前区间存在最小值为0的点，则表示存在解，若最后都未找到解，则为NO。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int ,int> pii;
#define endl '\n'
ll gcd(ll a, ll b){
    return b == 0 ? a : gcd(b, a % b);
}
void input(){
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
}
const int N = 4e6+10, M = N * 2, inf = 1e8;
// 一棵维护区间最小值的线段树，支持区间加
struct SegmentTree{
    int l, r, v, lazy;
}tr[N];
int n, d;
struct Line{
    int l, r, add;
};
vector<Line> G[N];
void pushup(int u){
    tr[u].v = min(tr[u<<1].v , tr[u<<1|1].v);
}
void pushdown(int u){
    if(tr[u].lazy){
        tr[u<<1].lazy += tr[u].lazy;
        tr[u<<1|1].lazy += tr[u].lazy;
        tr[u<<1].v += tr[u].lazy;
        tr[u<<1|1].v += tr[u].lazy;
        tr[u].lazy = 0;
    }
}
void build(int u, int l, int r){
    tr[u].l = l, tr[u].r = r, tr[u].v = 0, tr[u].lazy = 0;
    if(l == r) return;
    else{
        int mid = (l + r) >> 1;
        build(u<<1, l, mid);
        build(u<<1|1, mid+1, r);
        pushup(u);
    }
}
void update(int u, int l, int r, int v){
    if(l <= tr[u].l && tr[u].r <= r){
        tr[u].lazy += v;
        tr[u].v += v;
    }else{
        pushdown(u);
        int mid = (tr[u].l + tr[u].r) >> 1;
        if(l <= mid) update(u<<1, l, r, v);
        if(r > mid) update(u<<1|1, l, r, v);
        pushup(u);
    }
}
int flag = 0;
void query(int u, int x){
    if(flag) return;
    if(tr[u].l == tr[u].r){
        if(tr[u].v == 0){
            cout<<"YES"<<endl;
            cout<<x<<" "<<tr[u].l<<endl;
            flag = 1;
        }
    }else{
        pushdown(u);
        query(u<<1, x);
        query(u<<1|1, x);
    }
}
void add(int x1, int x2, int y1, int y2){
    //cout<<x1<<" "<<x2<<" "<<y1<<" "<<y2<<endl;
    G[x1].push_back({y1, y2, 1});
    G[x2+1].push_back({y1, y2, -1});
}
int main(){
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
  //  input();    
    cin>>n>>d;
    for(int i = 1; i <= n; i++){
        int x1, x2, y1, y2; cin>>x1>>y1>>x2>>y2;
        x2 --, y2 --;
        if(x2 - x1 + 1 >= d) x1 = 0, x2 = d - 1;
        if(y2 - y1 + 1 >= d) y1 = 0, y2 = d - 1;
        x1 = (x1 % d + d) % d, x2 = (x2 % d + d) % d, y1 = (y1 % d + d) % d, y2 = (y2 % d + d) % d;
        if(x2 >= x1){
            if(y2 >= y1){
                add(x1, x2, y1, y2);
            }else{
                add(x1, x2, 0, y2);
                add(x1, x2, y1, d-1);
            }
        }else{
            if(y2 >= y1){
                add(0, x2, y1, y2);
                add(x1, d-1, y1, y2);
            }else{
                add(0, x2, 0, y2);
                add(0, x2, y1, d-1);
                add(x1, d-1, 0, y2);
                add(x1, d-1, y1, d-1);
            }
        }
    }
    build(1, 0, d-1);
    for(int i = 0; i < d; i++){
        for(auto j : G[i]) update(1, j.l, j.r, j.add);
        if(tr[1].v == 0) query(1, i);
        if(flag) return 0;
    }
    cout<<"NO"<<endl;
    return 0;
}