#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(int s, int t);
mt19937 mt(3);
long long n, m;
vector<int> adj[MAXN], idx[MAXN], w;
bool used[MAXN];
long long a, b;
long long sm;
queue<long long> q;
vector<pair<long long, long long> > bfs(long long x){
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;
vector<int> temp;
for(int i = 0; i < (int)adj[u].size(); i++){
temp.push_back(i + 1);
}
for(int i = 0; i < (int)adj[u].size(); i++){
int x = mt() % adj[u].size();
int y = mt() % adj[u].size();
swap(temp[x], temp[y]);
}
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);
}
}
for(long long i = 0; i < adj[u].size(); i++){
long long to = adj[u][temp[i]];
long long j = idx[u][temp[i]];
if(!used[to]){
v.push_back({j, to});
}
}
}
return v;
}
long long away(int x, const vector<int> &U, const vector<int> &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(int N, vector<int> U, vector<int> V, int A, int 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
*/
Compilation message
highway.cpp: In function 'std::vector<std::pair<long long int, long long int> > bfs(long long int)':
highway.cpp:53:26: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(long long i = 0; i < adj[u].size(); i++){
~~^~~~~~~~~~~~~~~
highway.cpp:55:14: warning: unused variable 'j' [-Wunused-variable]
long long j = idx[u][i];
^
highway.cpp:62:26: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(long long i = 0; i < adj[u].size(); i++){
~~^~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
13 ms |
10024 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
13 ms |
10100 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
25 ms |
11680 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
14 ms |
10180 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
36 ms |
11780 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
28 ms |
11784 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |