This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include"bits/stdc++.h"
#include "stations.h"
#include <vector>
using namespace std;
const int N = 1000;
int T;
vector<int> adj[N], lb(N);
vector<bool> vis(N);
void dfs(int s, bool b){ //b = 0 apertura, b = 1 chiusura
if(vis[s]) return;
vis[s] = 1;
if(!b) lb[s] = T; T++;
for(int u : adj[s]) dfs(u, 1-b);
if(b) lb[s] = T; T++;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
for(int i=0; i<n; ++i) adj[i].clear();
lb.resize(n);
fill(lb.begin(), lb.end(), -1);
fill(vis.begin(), vis.end(), 0);
T = 0;
for(int i=0; i<n-1; ++i){
adj[u[i]].push_back(v[i]);
adj[v[i]].push_back(u[i]);
}
dfs(0, 0);
//for(int i=0; i<n; ++i) cout << i << " " << lb[i] << endl;
map<int, int> mp;
for(int i=0; i<n; ++i) mp[lb[i]] = i;
sort(lb.begin(), lb.end());
vector<int> lb2(n);
for(int i=0; i<n; ++i){
int j = mp[lb[i]];
lb2[j] = i;
}
return lb2;
}
int find_next_station(int s, int t, vector<int> c) {
int m=0;
for(int u : c) if(u>s) m++;
int par=-1;
if(m == c.size()){ //s apertura, le altre sono tutte chiusure
par = 0;
for(int u : c) par = max(par, u);
if(t <= s) return par;
for(int i=0; i<c.size(); ++i) if(c[i] >= t) return c[i];
} else{ // s chiusura, le altre sono tutte aperture
par = 400001;
for(int u : c) par = min(par, u);
if(t >= s) return par;
for(int i=c.size()-1; i>=0; --i) if(c[i] <= t) return c[i];
}
return par;
}
Compilation message (stderr)
stations.cpp: In function 'void dfs(int, bool)':
stations.cpp:15:2: warning: this 'if' clause does not guard... [-Wmisleading-indentation]
15 | if(!b) lb[s] = T; T++;
| ^~
stations.cpp:15:20: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'if'
15 | if(!b) lb[s] = T; T++;
| ^
stations.cpp:17:2: warning: this 'if' clause does not guard... [-Wmisleading-indentation]
17 | if(b) lb[s] = T; T++;
| ^~
stations.cpp:17:19: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'if'
17 | if(b) lb[s] = T; T++;
| ^
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:50:7: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
50 | if(m == c.size()){ //s apertura, le altre sono tutte chiusure
| ~~^~~~~~~~~~~
stations.cpp:54:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
54 | for(int i=0; i<c.size(); ++i) if(c[i] >= t) return c[i];
| ~^~~~~~~~~| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |