#include <bits/stdc++.h>
#include "library.h"
using namespace std;
vector<vector<int>> adj;
vector<int> comp;
vector<bool> vis;
vector<int> m;
void init(int n){
for (int i = 0; i < n; i++) m[i] = 0;
}
bool check(int i, int j, int n){
if (i == j) return false;
for (int k = i; k <= j; k++){
m[k] = 1;
}
int cnt = 0;
for (auto x : adj[i]) m[x] = 0;
for (int i = 0; i < n; i++) cnt += m[i];
if (cnt == 1) {
init(n);
return false;
}
int a1 = Query(m);
m[i] = 0;
int a2;
if (cnt != 2)
a2 = Query(m);
else a2 = 1;
init(n);
return a2 >= a1;
}
void dfs(int u, int par = -1){
comp.push_back(u + 1);
vis[u] = true;
for (int v : adj[u]){
if (v != par){
assert(!vis[v]);
dfs(v, u);
}
}
}
void Solve(int n)
{
// vector<int> M(N);
// for(int i = 0; i < N; i++) {
// M[i] = 1;
// }
// int A = Query(M);
// vector<int> res(N);
// for(int i = 0; i < N; i++) {
// res[i] = i + 1;
// }
// Answer(res);
if (n == 1){
Answer({1});
return;
}
if (n == 2){
Answer({1, 2});
return;
}
m.resize(3 * n, 0);
adj.resize(3 * n);
vis.resize(3 * n, false);
int ok = 0;
for (int i = 0; i < n; i++){
while (adj[i].size() < 2 && check(i, n - 1, n)){
int l = i + 1, r = n - 1;
while (l != r){
int m = (l + r)/2;
if (check(i, m, n)) r = m;
else l = m + 1;
}
ok++;
adj[i].push_back(l);
adj[l].push_back(i);
}
}
// assert(ok == n - 1);
int holy = 0;
for (int i = 0; i < n; i++) {
holy += adj[i].size();
// assert(adj[i].size() >= 1);
}
// assert(holy == 2 * n - 2);
for (int i = 0; i < n; i++){
if (adj[i].size() == 1){
dfs(i);
break;
}
}
// assert(comp.size() == n);
Answer(comp);
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
208 KB |
Wrong Answer [1] |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
208 KB |
Wrong Answer [1] |
2 |
Halted |
0 ms |
0 KB |
- |