Submission #1096440

# Submission time Handle Problem Language Result Execution time Memory
1096440 2024-10-04T12:44:51 Z Ejen Park (BOI16_park) C++17
100 / 100
257 ms 52816 KB
#include <bits/stdc++.h>
 
#define el "\n"
#define fu(i, a, b) for (int i = a; i <= b; ++i)
#define fd(i, a, b) for (int i = a; i >= b; --i)
#define ff first
#define ss second
#define all(v) (v).begin(), (v).end()
#define sz(v) (ll)(v).size()
#define mask(i) (1LL << (i))
#define bit(x, i) ((x) >> (i) & 1)
 
using namespace std;
 
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
 
mt19937_64 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
 
ll Rand(ll l, ll r) {
    return uniform_int_distribution<ll>(l, r) (rng);
}
 
ll last(ll msk)     {return msk & (-msk);}
ll pop_cnt(ll msk)  {return __builtin_popcountll(msk);}
ll ctz(ll msk)      {return __builtin_ctzll(msk);}
ll lg(ll msk)       {return 63 - __builtin_clzll(msk);}
 
template<class T1, class T2> bool minimize(T1 &a, T2 b) {
    return a > b ? a = b, 1 : 0;
}
 
template<class T1, class T2> bool maximize(T1 &a, T2 b) {
    return a < b ? a = b, 1 : 0;
}
 
template<class T> void print(T a) {
    for (auto x : a) cout << x << " ";
    cout << el;
}
 
template<class T> void compress(T &a) {
    sort(all(a));
    a.resize(unique(all(a)) - a.begin());
}
 
const long long N = 1e5 + 27, M = 1e5 + 27, inf = 2e18, bl = 320, base = 311, mod = 123456789, lim = 1e6;
 
struct DSU {
    ll n;
    vector<ll> lab;
 
    DSU(ll _n) {
        n = _n;
        lab.resize(n + 27, -1);
    }
 
    ll find(ll u) {
        return lab[u] < 0 ? u : lab[u] = find(lab[u]);
    }
 
    bool join(ll u, ll v) {
        u = find(u);
        v = find(v);
        if (u == v) return false;
        if (lab[u] > lab[v]) swap(u, v);
        lab[u] += lab[v];
        lab[v] = u;
        return true;
    }
 
    bool connect(ll u, ll v) {
        u = find(u);
        v = find(v);
        return u == v;
    }
};
 
struct Edge {
    ll u, v, w;
    Edge(ll _u, ll _v, ll _w) {
        u = _u, v = _v, w = _w;
    }
    bool operator < (const Edge &other) const {
        return w < other.w;
    }
};
 
ll n, q, w, h;
ll x[N], y[N], r[N];
bool ans[5][N];
pair<ll, ll> tourist[N];
vector<ll> group[5];
vector<Edge> edge;
 
ll sq(ll x) {
    return x * x;
}
 
ll dist(ll i, ll j) { 
    return sqrt(sq(x[i] - x[j]) + sq(y[i] - y[j])) - r[i] - r[j];
}
 
signed main() {
    // freopen("bdbang.inp", "r", stdin);
    // freopen("bdbang.out", "w", stdout);
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> q >> w >> h;
    fu(i, 1, n) cin >> x[i] >> y[i] >> r[i];
    fu(i, 1, n) {
        edge.push_back(Edge(n + 1, i, x[i] - r[i]));
        edge.push_back(Edge(n + 2, i, y[i] - r[i]));
        edge.push_back(Edge(n + 3, i, w - x[i] - r[i]));
        edge.push_back(Edge(n + 4, i, h - y[i] - r[i]));
        fu(j, i + 1, n) edge.push_back(Edge(i, j, dist(i, j)));
    }
    sort(all(edge));
    // cout << dist(1, 2);
    fu(i, 1, q) {
        cin >> tourist[i].ss >> tourist[i].ff;
        tourist[i].ss *= 2;
        group[tourist[i].ff].push_back(i);
    }
    {
        DSU dsu(n + 4);
        vector<pair<ll, ll>> a;
        for (ll id : group[1]) a.push_back(make_pair(tourist[id].ss, id));
        
        sort(all(a));
        ll j = 0;
        fu(i, 0, sz(a) - 1) {
            while (j < sz(edge) && edge[j].w < a[i].ff) {
                dsu.join(edge[j].u, edge[j].v);
                j++;
            }
            ans[1][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 3) && !dsu.connect(n + 1, n + 2) && !dsu.connect(n + 4, n + 1)) ans[4][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 2) && !dsu.connect(n + 4, n + 2) && !dsu.connect(n + 3, n + 2)) ans[2][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 2) && !dsu.connect(n + 4, n + 2) && !dsu.connect(n + 4, n + 3) && !dsu.connect(n + 1, n + 3)) ans[3][a[i].ss] = true;
        }
    }
 
    {
        DSU dsu(n + 4);
        vector<pair<ll, ll>> a;
        for (ll id : group[2]) a.push_back(make_pair(tourist[id].ss, id));
        sort(all(a));
        ll j = 0;
        fu(i, 0, sz(a) - 1) {
            while (j < sz(edge) && edge[j].w < a[i].ff) {
                dsu.join(edge[j].u, edge[j].v);
                j++;
            }  
            ans[2][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 3) && !dsu.connect(n + 4, n + 2) && !dsu.connect(n + 2, n + 3) && !dsu.connect(n + 1, n + 4)) ans[4][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 2) && !dsu.connect(n + 4, n + 2) && !dsu.connect(n + 2, n + 3)) ans[1][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 3) && !dsu.connect(n + 3, n + 2) && !dsu.connect(n + 4, n + 3)) ans[3][a[i].ss] = true;
 
        }
    }
    {
        DSU dsu(n + 4);
        vector<pair<ll, ll>> a;
        for (ll id : group[3]) a.push_back(make_pair(tourist[id].ss, id));
        sort(all(a));
        ll j = 0;
        fu(i, 0, sz(a) - 1) {
            while (j < sz(edge) && edge[j].w < a[i].ff) {
                dsu.join(edge[j].u, edge[j].v);
                j++;
            }  
            ans[3][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 2) && !dsu.connect(n + 4, n + 2) && !dsu.connect(n + 4, n + 3) && !dsu.connect(n + 1, n + 3)) ans[1][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 3) && !dsu.connect(n + 3, n + 2) && !dsu.connect(n + 4, n + 3)) ans[2][a[i].ss] = true;
            if (!dsu.connect(n + 4, n + 2) && !dsu.connect(n + 4, n + 1) && !dsu.connect(n + 4, n + 3)) ans[4][a[i].ss] = true;
 
        }
    }
    {
        DSU dsu(n + 4);
        vector<pair<ll, ll>> a;
        for (ll id : group[4]) a.push_back(make_pair(tourist[id].ss, id));
        sort(all(a));
        ll j = 0;
        fu(i, 0, sz(a) - 1) {
            while (j < sz(edge) && edge[j].w < a[i].ff) {
                dsu.join(edge[j].u, edge[j].v);
                j++;
            }  
            ans[4][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 3) && !dsu.connect(n + 1, n + 2) && !dsu.connect(n + 4, n + 1)) ans[1][a[i].ss] = true;
            if (!dsu.connect(n + 1, n + 3) && !dsu.connect(n + 4, n + 2) && !dsu.connect(n + 2, n + 3) && !dsu.connect(n + 1, n + 4)) ans[2][a[i].ss] = true;
            if (!dsu.connect(n + 4, n + 2) && !dsu.connect(n + 4, n + 1) && !dsu.connect(n + 4, n + 3)) ans[3][a[i].ss] = true;
        }
    }
    fu(i, 1, q) {
        fu(j, 1, 4) if (ans[j][i]) cout << j; 
        cout << el;
    }
 
 
}
# Verdict Execution time Memory Grader output
1 Correct 219 ms 49832 KB Output is correct
2 Correct 207 ms 49852 KB Output is correct
3 Correct 235 ms 49852 KB Output is correct
4 Correct 218 ms 49960 KB Output is correct
5 Correct 213 ms 49796 KB Output is correct
6 Correct 213 ms 49824 KB Output is correct
7 Correct 213 ms 49820 KB Output is correct
8 Correct 196 ms 49824 KB Output is correct
9 Correct 0 ms 344 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 34 ms 5980 KB Output is correct
2 Correct 32 ms 6004 KB Output is correct
3 Correct 31 ms 5952 KB Output is correct
4 Correct 33 ms 6208 KB Output is correct
5 Correct 34 ms 6220 KB Output is correct
6 Correct 36 ms 6216 KB Output is correct
7 Correct 34 ms 5280 KB Output is correct
8 Correct 30 ms 5232 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 219 ms 49832 KB Output is correct
2 Correct 207 ms 49852 KB Output is correct
3 Correct 235 ms 49852 KB Output is correct
4 Correct 218 ms 49960 KB Output is correct
5 Correct 213 ms 49796 KB Output is correct
6 Correct 213 ms 49824 KB Output is correct
7 Correct 213 ms 49820 KB Output is correct
8 Correct 196 ms 49824 KB Output is correct
9 Correct 0 ms 344 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 34 ms 5980 KB Output is correct
12 Correct 32 ms 6004 KB Output is correct
13 Correct 31 ms 5952 KB Output is correct
14 Correct 33 ms 6208 KB Output is correct
15 Correct 34 ms 6220 KB Output is correct
16 Correct 36 ms 6216 KB Output is correct
17 Correct 34 ms 5280 KB Output is correct
18 Correct 30 ms 5232 KB Output is correct
19 Correct 249 ms 52816 KB Output is correct
20 Correct 251 ms 52816 KB Output is correct
21 Correct 244 ms 52796 KB Output is correct
22 Correct 248 ms 52504 KB Output is correct
23 Correct 246 ms 52560 KB Output is correct
24 Correct 257 ms 52640 KB Output is correct