Submission #1191087

#	Time	Username	Problem	Language	Result	Execution time	Memory
1191087		Malix	Circle selection (APIO18_circle_selection)	C++20	87 / 100	3038 ms	473076 KiB

circle_selection

#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vii;
typedef pair<int,int> pi;
typedef vector<pi> pii;
typedef tuple<ll,ll,ll> ti;
typedef vector<ll> li;
typedef vector<li> lii;
 
#define REP(i,a,b) for(int i=a;i<b;i++)
#define F first
#define S second
#define PB push_back
#define LSOne(s) ((s)&(-s))
 
ll INF=1000000000000000010;
int inf=1e9+10;
ll M=1e9+7;

vector<ti> a;
set<tuple<ll,ll,int>> st;
unordered_map<ll,vector<int>> mp;
vi ans,arr;
ll sz,ml=2000000001;
int n;

bool ok(int x,int y){
    ll x1=get<0>(a[x]);
    ll y1=get<1>(a[x]);
    ll z1=get<2>(a[x]);
    ll x2=get<0>(a[y]);
    ll y2=get<1>(a[y]);
    ll z2=get<2>(a[y]);
    return ((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)<=(z1+z2)*(z1+z2));
}

void change(){
    sz/=2;
    mp.clear();
    REP(i,0,n)if(!arr[i]){
        ll x=get<0>(a[i]);
        ll y=get<1>(a[i]);
        mp[x/sz+(y/sz)*ml].PB(i);
    }
}

int main() {   
// ios::sync_with_stdio(0);
// cin.tie(0);
    cin>>n;
    a.resize(n);arr.resize(n,0);
    priority_queue<pair<ll,int>> pq;
    ll mx=0;
    ans.resize(n);
    REP(i,0,n)ans[i]=i;
    REP(i,0,n){
        ll x,y,z;cin>>x>>y>>z;
        a[i]={x,y,z};
        pq.push({z,-i});
        mx=max(mx,z);
    }
    int t=log2(mx)+2;
    sz=(1LL<<t);
    change();
    ll pos=0;
    while(!pq.empty()){
        ll r=pq.top().F;
        int pos=-pq.top().S;
        pq.pop();
        arr[pos]=1;
        if(ans[pos]!=pos)continue;
        if(r<=sz/2)change();
        ll x=get<0>(a[pos]);
        ll y=get<1>(a[pos]);
        st.erase({x,y,pos});
        x-=x%sz+2*sz;
        y-=y%sz+2*sz;
        x/=sz;y/=sz;
        REP(i,0,5){
            REP(j,0,5){
                for(auto u:mp[(x+i)+(y+j)*ml])if(arr[u]==0&&ok(pos,u)){
                    arr[u]=1;
                    ans[u]=pos;
                }
            }
        }
    }
    REP(i,0,n)cout<<ans[i]+1<<" ";
}

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