# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
121483 | 2019-06-26T14:33:12 Z | Sorting | Highway Tolls (IOI18_highway) | C++14 | 0 ms | 0 KB |
#include <bits/stdc++.h> #include "highway.h" using namespace std; const long long MAXN = 1e5 + 7; long long ask(const vector<int> &w); void answer(long long s, long long t); mt19937 mt(); long long n, m; vector<int> adj[MAXN], idx[MAXN], w; bool used[MAXN]; long long a, b; long long sm; vector<pair<long long, long long> > bfs(long long x){ queue<long long> q; for(long long i = 0; i < n; i++){ used[i] = false; } q.push(x); vector<pair<long long, long long> > v; while(!q.empty()){ long long u = q.front(); q.pop(); if(used[u]){ continue; } used[u] = true; for(long long i = 0; i < adj[u].size(); i++){ long long to = adj[u][i]; long long j = idx[u][i]; if(!used[to]){ q.push(to); v.push_back({j, to}); } } } return v; } long long away(long long x, const vector<long long> &U, const vector<long long> &V){ vector<pair<long long, long long> > v = bfs(x); while((long long)v.size() != m); long long l = 0, r = (long long)v.size() - 1; while(l != r){ long long mid = (l + r) >> 1; for(long long i = 0; i < m; i++){ w[i] = 0; } for(long long i = 0; i <= mid; i++){ w[ v[i].first ] = 1; } long long score = ask(w); if(score == (sm / a) * b){ r = mid; } else{ l = mid + 1; } } return v[l].second; } void find_pair(long long N, vector<long long> U, vector<long long> V, long long A, long long B){ srand(time(NULL)); n = N; m = U.size(); for(long long i = 0; i < m; i++){ adj[V[i]].push_back(U[i]); adj[U[i]].push_back(V[i]); idx[V[i]].push_back(i); idx[U[i]].push_back(i); w.push_back(0); } sm = ask(w); a = A; b = B; long long l = 0, r = m - 1; while(l != r){ long long mid = (l + r) / 2; for(long long i = 0; i < m; i++){ w[i] = 0; } for(long long i = mid + 1; i <= r; i++){ w[i] = 1; } long long score = ask(w); score -= sm; if(score){ l = mid + 1; } else{ r = mid; } } long long x, y; x = away(U[l], U, V); y = away(x, U, V); answer(x, y); } /* 9 1 2 4 8 4 2 8 0 2 5 2 7 2 3 0 1 0 6 */