Submission #777235

#TimeUsernameProblemLanguageResultExecution timeMemory
777235Sam_a17Circle selection (APIO18_circle_selection)C++17
23 / 100
1275 ms50464 KiB
#define _CRT_SECURE_NO_WARNINGS #include <bits/stdc++.h> //#include "temp.cpp" #include <cstdio> using namespace std; #ifndef ONLINE_JUDGE #define dbg(x) cerr << #x <<" "; print(x); cerr << endl; #else #define dbg(x) #endif #define sz(x) (int)x.size() #define len(x) (int)x.length() #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define clr(x) (x).clear() #define uniq(x) x.resize(unique(all(x)) - x.begin()); #define blt __builtin_popcount #define pb push_back #define popf pop_front #define popb pop_back #define ld long double #define ll long long void print(long long t) {cerr << t;} void print(int t) {cerr << t;} void print(string t) {cerr << t;} void print(char t) {cerr << t;} void print(double t) {cerr << t;} void print(long double t) {cerr << t;} void print(unsigned long long t) {cerr << t;} #include <ext/pb_ds/assoc_container.hpp> using namespace __gnu_pbds; #define nl '\n' // Indexed Set template <class T> using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>; template <class T, class V> void print(pair <T, V> p); template <class T> void print(vector <T> v); template <class T> void print(set <T> v); template <class T, class V> void print(map <T, V> v); template <class T> void print(multiset <T> v); template <class T, class V> void print(T v[],V n) {cerr << "["; for(int i = 0; i < n; i++) {print(v[i]); cerr << " "; } cerr << "]";} template <class T, class V> void print(pair <T, V> p) {cerr << "{"; print(p.first); cerr << ","; print(p.second); cerr << "}";} template <class T> void print(vector <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} // template <class T> void print(vector <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(set <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(multiset <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(Tree <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} template <class T, class V> void print(map <T, V> v) {cerr << "[ "; for (auto i : v) {print(i); cerr << " ";} cerr << "]";} template <class T> void print(deque <T> v) {cerr << "[ "; for (T i : v) {print(i); cerr << " ";} cerr << "]";} // for random generations mt19937 myrand(chrono::steady_clock::now().time_since_epoch().count()); // mt19937 myrand(131); // for grid problems int dx[8] = {-1,0,1,0,1,-1,1,-1}; int dy[8] = {0,1,0,-1,1,1,-1,-1}; // lowest / (1 << 17) >= 1e5 / (1 << 18) >= 2e5 / (1 << 21) >= 1e6 void fastIO() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr); } // file in/out void setIO(string str = "") { fastIO(); // if(str == "input") { // freopen("input.txt", "r", stdin); // freopen("output.txt", "w", stdout); // } else if(str != "") { // freopen((str + ".in").c_str(), "r", stdin); // freopen((str + ".out").c_str(), "w", stdout); // } } const long long inf = 1e9 + 1; int n; long long dist(pair<long long, long long> a, pair<long long, long long> b) { return abs(a.first - b.first) * abs(a.first - b.first) + abs(a.second - b.second) * abs(a.second - b.second); } struct node { long long x, y, r; int index; }; map<pair<long long, long long>, vector<node>> mp; bool cmp(node a, node b) { if(a.r > b.r) { return true; } else { if(a.index < b.index) { return true; } else { return false; } } } void solve_() { cin >> n; vector<node> a; long long cur_r, cur; for(int i = 1; i <= n; i++) { long long x, y, r; cin >> x >> y >> r; x += inf, y += inf; a.push_back({x, y, r, i}); } sort(all(a), cmp); vector<int> which(n + 1); long long mx = -1; for(int w = 0; w < n; w++) { int ind = a[w].index; if(which[ind]) continue; if(mx == -1 || mx / 2 > a[w].r) { mx = a[w].r; mp.clear(); mx = max(mx, 1ll); for(int i = 0; i < n; i++) { if(which[a[i].index]) { continue; } long long cx = a[i].x / mx; long long cy = a[i].y / mx; mp[{cx, cy}].push_back(a[i]); } } long long cur_x = a[w].x / mx; long long cur_y = a[w].y / mx; for(int i = cur_x - 2; i <= cur_x + 2; i++) { for(int j = cur_y - 2; j <= cur_y + 2; j++) { if(mp.find({i, j}) == mp.end()) continue; for(auto k: mp[{i, j}]) { if(which[k.index]) continue; if(dist({a[w].x, a[w].y}, {k.x, k.y}) <= (a[w].r + k.r) * (a[w].r + k.r)) { which[k.index] = ind; } } } } } for(int i = 1; i <= n; i++) { assert(which[i]); cout << which[i] << " "; } cout << '\n'; } int main() { setIO(); auto solve = [&](int test_case)-> void { for(int i = 1; i <= test_case; i++) { solve_(); } }; int test_cases = 1; // cin >> test_cases; solve(test_cases); return 0; }

Compilation message (stderr)

circle_selection.cpp: In function 'void solve_()':
circle_selection.cpp:114:13: warning: unused variable 'cur_r' [-Wunused-variable]
  114 |   long long cur_r, cur;
      |             ^~~~~
circle_selection.cpp:114:20: warning: unused variable 'cur' [-Wunused-variable]
  114 |   long long cur_r, cur;
      |                    ^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...