Submission #49135

#	Time	Username	Problem	Language	Result	Execution time	Memory
49135		BThero	Circle selection (APIO18_circle_selection)	C++17	0 / 100	3021 ms	41320 KiB

circle_selection

// Why am I so stupid? :c
#include <bits/stdc++.h>

#define pb push_back
#define mp make_pair

#define all(x) (x).begin(), (x).end()

#define fi first
#define se second

typedef long long ll;

using namespace std;

struct circle {
    int x, y, r;

    circle() {
        x = y = r = 0;
    }
} arr[300005];

map < pair <int, int>, vector <int> > mm;

int ans[300005];

int u[300005];

int n, cur;

bool inter(circle a, circle b) {
    ll dist = (a.x - b.x) * 1ll * (a.x - b.x) + (a.y - b.y) * 1ll * (a.y - b.y);
    return dist <= a.r + b.r;
}

void solve() {
    scanf("%d", &n);

    for (int i = 1; i <= n; ++i) {
        scanf("%d %d %d", &arr[i].x, &arr[i].y, &arr[i].r);
    }

    int R = arr[1].r;

    for (int i = 1; i <= n; ++i) {
        mm[mp(arr[i].x / R, arr[i].y / R)].pb(i);
        ans[i] = i;
    }

    vector <int> vv;

    for (int i = 1; i <= n; ++i) {
        if (ans[i] != i) {
            continue;
        }

        int cx = arr[i].x / R, cy = arr[i].y / R;

        for (int x = cx - 2; x <= cx + 2; ++x) {
            for (int y = cy - 2; y <= cy + 2; ++y) {
                if (mm.find(mp(x, y)) == mm.end()) {
                    continue;
                }

                vv.clear();
                ++cur;

                for (int it : mm[mp(x, y)]) {
                    if (inter(arr[i], arr[it])) {
                        u[it] = cur;
                        ans[it] = i;
                        vv.pb(it);
                    }
                }

                for (int it : mm[mp(x, y)]) {
                    if (u[it] != cur) {
                        vv.pb(it);
                    }
                }

                mm[mp(x, y)] = vv;
            }
        }
    }

    for (int i = 1; i <= n; ++i) {
        printf("%d ", ans[i]);
    }
}

int main() {
#ifdef BThero
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
#endif // BThero

    int tt = 1;

    while (tt--) {
        solve();
    }

    return 0;
}

Compilation message (stderr)

circle_selection.cpp: In function 'void solve()':
circle_selection.cpp:38:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d", &n);
     ~~~~~^~~~~~~~~~
circle_selection.cpp:41:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d %d %d", &arr[i].x, &arr[i].y, &arr[i].r);
         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 7
• Subtask 2: 0 / 12
• Subtask 3: 0 / 15
• Subtask 4: 0 / 23
• Subtask 5: 0 / 30
• Subtask 6: 0 / 13