#include <bits/stdc++.h>
#include "stations.h"
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
#define N 100005
#define PB push_back
#define sz(x) int(x.size())
#define F first
#define M ll(1e9 + 7)
#define S second
#define all(x) x.begin(), x.end()
#define endl '\n'
//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("-O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("fast-math")
//#pragma GCC optimize("no-stack-protector")
using namespace std;
//using namespace __gnu_pbds;
typedef long long ll;
//typedef tree <int, null_type, less_equal <int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;
ll mlt(ll x, ll y) {return (x * y) % M;}
ll sm(ll x, ll y) {return (x + y) % M;}
vector <int> g[N], a;
int id = 0;
void dfs(int v, int p)
{
a[v] = id++;
for (auto it : g[v])
{
if (it == p) continue;
dfs(it, v);
}
}
void rec(int v, int p)
{
for (auto it : g[v])
{
if (it == p) continue;
rec(it, v);
}
a[v] = id++;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v)
{
a.resize(n);
id = 0;
for (int i = 0; i < n - 1; i++)
{
g[u[i]].PB(v[i]);
g[v[i]].PB(u[i]);
}
bool f = 1;
for (int i = 0; i < n; i++) if (sz(g[i]) > 2) f = 0;
if (f)
{
for (int i = 0; i < n; i++)
if (sz(g[i]) == 1) {dfs(i, -1); break;}
}
else rec(0, -1);
for (int i = 0; i < n; i++) g[i].clear();
return a;
}
int find_next_station(int s, int t, vector<int> c)
{
if (sz(c) == 1) return c[0];
if (sz(c) == 2)
{
sort(all(c));
if (t < c[1]) return c[0];
return c[1];
}
if (sz(c) == 3)
{
sort(all(c));
if (t < c[1]) return c[0];
if (t < c[2]) return c[1];
return c[2];
}
}
//int main()
//{
// iostream::sync_with_stdio(0); ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
//
// //freopen("fcolor.in", "r", stdin); freopen("fcolor.out", "w", stdout);
//
//
//}
Compilation message
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:111:1: warning: control reaches end of non-void function [-Wreturn-type]
111 | }
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
534 ms |
5740 KB |
Output is correct |
2 |
Correct |
454 ms |
5616 KB |
Output is correct |
3 |
Correct |
1085 ms |
5840 KB |
Output is correct |
4 |
Correct |
722 ms |
5376 KB |
Output is correct |
5 |
Correct |
720 ms |
5664 KB |
Output is correct |
6 |
Correct |
450 ms |
5732 KB |
Output is correct |
7 |
Correct |
445 ms |
5632 KB |
Output is correct |
8 |
Correct |
6 ms |
5724 KB |
Output is correct |
9 |
Correct |
7 ms |
5376 KB |
Output is correct |
10 |
Correct |
5 ms |
5504 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
451 ms |
5504 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
541 ms |
5732 KB |
Output is correct |
2 |
Correct |
506 ms |
5732 KB |
Output is correct |
3 |
Correct |
920 ms |
5596 KB |
Output is correct |
4 |
Correct |
624 ms |
5504 KB |
Output is correct |
5 |
Correct |
581 ms |
5760 KB |
Output is correct |
6 |
Correct |
438 ms |
5752 KB |
Output is correct |
7 |
Correct |
523 ms |
5752 KB |
Output is correct |
8 |
Correct |
7 ms |
5720 KB |
Output is correct |
9 |
Correct |
7 ms |
5596 KB |
Output is correct |
10 |
Correct |
4 ms |
5664 KB |
Output is correct |
11 |
Correct |
564 ms |
5724 KB |
Output is correct |
12 |
Incorrect |
454 ms |
5860 KB |
Wrong query response. |
13 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
918 ms |
5592 KB |
Output is correct |
2 |
Correct |
793 ms |
5712 KB |
Output is correct |
3 |
Correct |
596 ms |
5588 KB |
Output is correct |
4 |
Correct |
6 ms |
5600 KB |
Output is correct |
5 |
Correct |
7 ms |
5592 KB |
Output is correct |
6 |
Correct |
4 ms |
5596 KB |
Output is correct |
7 |
Correct |
583 ms |
5724 KB |
Output is correct |
8 |
Correct |
935 ms |
5504 KB |
Output is correct |
9 |
Correct |
753 ms |
5588 KB |
Output is correct |
10 |
Correct |
721 ms |
5592 KB |
Output is correct |
11 |
Incorrect |
8 ms |
5504 KB |
Wrong query response. |
12 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
611 ms |
5732 KB |
Output is correct |
2 |
Correct |
492 ms |
5632 KB |
Output is correct |
3 |
Correct |
955 ms |
5588 KB |
Output is correct |
4 |
Correct |
828 ms |
5504 KB |
Output is correct |
5 |
Correct |
659 ms |
5632 KB |
Output is correct |
6 |
Correct |
535 ms |
5732 KB |
Output is correct |
7 |
Correct |
456 ms |
5888 KB |
Output is correct |
8 |
Correct |
5 ms |
5712 KB |
Output is correct |
9 |
Correct |
7 ms |
5632 KB |
Output is correct |
10 |
Correct |
4 ms |
5504 KB |
Output is correct |
11 |
Incorrect |
491 ms |
5516 KB |
Wrong query response. |
12 |
Halted |
0 ms |
0 KB |
- |