#include <bits/stdc++.h>
#include "stations.h"
using namespace std;
bool line = 1;
bool binary = 1;
const int N = 1001;
vector<vector<int>> go(N, vector<int>(N));
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
vector<vector<int>> adj(n);
for (int i = 0; i < n-1; i++) {
if (u[i] > v[i]) swap(u[i], v[i]);
binary &= (v[i] + 1) / 2 == (u[i] + 1);
adj[u[i]].push_back(v[i]);
adj[v[i]].push_back(u[i]);
}
/*
int r = -1;
for (int i = 0; i < n; i++) {
if (adj[i].size() == 1) r = i;
line &= adj[i].size() <= 2;
}
if (line) {
vector<int> ans(n);
int t = 0;
auto dfs = [&](auto&& s, int v, int p) -> void {
ans[v] = t++;
for (int u : adj[v]) if (u != p) s(s, u, v);
};
dfs(dfs, r, -1);
return ans;
}
*/
if (binary) {
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) go[i][j] = -1;
auto dfs = [&](auto&& s, int v) -> void {
for (int u : adj[v]) if (u != go[i][v]) {
go[i][u] = v;
s(s, u);
}
};
dfs(dfs, i);
}
vector<int> ans(n);
iota(ans.begin(), ans.end(), 0);
return ans;
}
/*
vector<int> tin(n), tout(n);
int timer = 0;
auto dfs = [&](auto&& s, int v, int p) -> void {
tin[v] = timer++;
for (int u : adj[v]) if (u != p) {
s(s, u, v);
}
tout[v] = timer - 1;
};
dfs(dfs, 0, -1);
vector<int> ans(n);
for (int i = 0; i < n; i++) ans[i] = tin[i] * 1000 + tout[i];
return ans;
*/
}
bool anc(int x, int y) {
int tins = x / 1000, touts = x % 1000;
int tint = y / 1000, toutt = y % 1000;
return tins <= tint && toutt <= touts;
}
int find_next_station(int s, int t, std::vector<int> c) {
/*
if (line) {
for (int x : c) if ((x-s)*(x-t) <= 0) return x;
}
*/
assert(binary);
if (binary) {
return go[t][s];
}
/*
if (anc(s, t)) {
for (int x : c) {
if (anc(s, x) && anc(x, t)) return x;
}
}
else {
for (int x : c) {
if (anc(x, s)) return x;
}
}
*/
assert(0);
}
Compilation message
stations.cpp: In function 'std::vector<int> label(int, int, std::vector<int>, std::vector<int>)':
stations.cpp:12:27: warning: control reaches end of non-void function [-Wreturn-type]
12 | vector<vector<int>> adj(n);
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
6 ms |
8536 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
301 ms |
8648 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
6 ms |
8792 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
521 ms |
8364 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
6 ms |
8536 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |