Submission #305712

#	Time	Username	Problem	Language	Result	Execution time	Memory
305712		peti1234	Stations (IOI20_stations)	C++17	100 / 100	1154 ms	1232 KiB

stations

#include <bits/stdc++.h>
#include "stations.h"
using namespace std;
const int c=1002;
vector<int> sz[c], sol;
bool v[c];
int cnt;
void dfs(int a, bool b) {
    v[a]=1;
    if (!b) sol[a]=cnt++;
    for (int i=0; i<sz[a].size(); i++) {
        int x=sz[a][i];
        if (!v[x]) dfs(x, !b);
    }
    if (b) sol[a]=cnt++;
}
vector<int> label(int n, int k, vector<int> x, vector<int> y) {
    for (int i=0; i<n; i++) sz[i].clear(), v[i]=0, cnt=0, sol.clear();
    sol.resize(n);
    for (int i=0; i<n-1; i++) sz[x[i]].push_back(y[i]), sz[y[i]].push_back(x[i]);
    dfs(0, 0);
    return sol;
}
int find_next_station(int s, int t, vector<int> sz) {
    int x=sz[0], y=sz.back(), si=sz.size();
    if (s<x) {
        if (t<s || t>y) return y;
        for (int i=0; i<si; i++) if (sz[i]>=t) return sz[i];
    }
    if (t>s || t<x) return x;
    for (int i=si-1; i>0; i--) if (sz[i]<=t) return sz[i];
}

Compilation message (stderr)

stations.cpp: In function 'void dfs(int, bool)':
stations.cpp:11:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   11 |     for (int i=0; i<sz[a].size(); i++) {
      |                   ~^~~~~~~~~~~~~
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:32:1: warning: control reaches end of non-void function [-Wreturn-type]
   32 | }
      | ^

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