Submission #104321

#	Time	Username	Problem	Language	Result	Execution time	Memory
104321		dimash241	Circle selection (APIO18_circle_selection)	C++17	0 / 100	2413 ms	47836 KiB

circle_selection

# include <stdio.h>
# include <bits/stdc++.h>
 
 
#define _USE_MATH_DEFINES_
#define ll long long
#define ld long double
#define Accepted 0
#define pb push_back
#define mp make_pair
#define sz(x) (int)(x.size())
#define every(x) x.begin(),x.end()
#define F first
#define S second
#define For(i,x,y)  for (int i = x; i <= y; i ++) 
#define FOr(i,x,y)  for (int i = x; i >= y; i --)
#define SpeedForce ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0)
// ROAD to...                                                                                                                                                                                                                Red
 
using namespace std;
inline double Time() {return (clock() * 1.0) / CLOCKS_PER_SEC; } 
inline bool isvowel (char c) {
	c = tolower(c);
    if (c == 'a' || c == 'e' || c == 'i' || c == 'y' || c == 'o' || c == 'u') return 1;
    return 0;
}
 
const double eps = 0.000001;
const ld pi = acos(-1);
const int maxn = 1e7 + 9;
const int mod = 1e9 + 7;
const ll MOD = 1e18 + 9;
const ll INF = 1e18 + 123;
const int inf = 2e9 + 11;
const int mxn = 1e6 + 9;
const int N = 3e5 + 123;                                          
const int M = 22;
const int pri = 997;
const int Magic = 2101;
 
const int dx[] = {-1, 0, 1, 0};
const int dy[] = {0, -1, 0, 1};
 
int n, m, k;
int id[N];
int r[N];
int x[N];
int y[N];
int ans[N];
int lim;

bool cmp (int x, int y) {
	return r[x] > r[y] || (r[x] == r[y] && x < y);
}
map < pair < int, int > , vector < int > > q;
void upd () {
	q.clear();
	for (int i = 1; i <= n; i ++) {
		if (!ans[i]) 
			q[mp(x[i] / lim, y[i] / lim)].pb(i);
	}
}
ll sq (ll x) {
	return x * x;
}

bool in (int i, int j) {
	return sq(x[i] - x[j]) + sq(y[i] - y[j]) <= sq(r[i] + r[j]);
}

int main () {
	SpeedForce;               

    cin >> n;
    for (int i = 1; i <= n; i ++) {
    	cin >> x[i] >> y[i] >> r[i];
    	id[i] = i;
    }

    sort(id + 1, id + n + 1, &cmp);
    lim = inf;
    for (int pos = 1; pos <= n; pos ++) {
    	if (ans[id[pos]]) continue;

    	int j = id[pos];
    	if (r[j] * 2 <= lim) {
    		lim = r[j];
    		upd ();
    	}
    //	cout << j << '\n';
    	ans[j] = j;
    	int xx = x[j] / lim, yy = y[j] / lim;
    	for (int x = xx - 2; x <= xx + 2; x ++) for (int y = yy - 2; y <= yy + 2; y ++) if (q.count(mp(x, y))) {
    		vector < int > nq;
    		for (auto i : q[mp(x, y)]) {
    			if (ans[i]) continue;
    			if (in(i, j)) {
    				ans[i] = j;
    			} else nq.pb(j);
    		}		
    		q[mp(x, y)] = nq;
    	}
    }
    for (int i = 1; i <= n; i ++)
    	cout << ans[i] << ' ';
    return Accepted;
}
 
// B...a D....

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