Submission #612292

#	Time	Username	Problem	Language	Result	Execution time	Memory
612292		daria	Stations (IOI20_stations)	C++17	100 / 100	994 ms	908 KiB

stations

#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];
      |                ~^~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 8 / 8
• Subtask 3: 16 / 16
• Subtask 4: 10 / 10
• Subtask 5: 61 / 61