Submission #586591

#	Time	Username	Problem	Language	Result	Execution time	Memory
586591		MohamedFaresNebili	Stations (IOI20_stations)	C++14	52.32 / 100	988 ms	840 KiB

stations

#include <bits/stdc++.h>
/// #include "stations.h"
/// #pragma GCC optimize ("Ofast")
/// #pragma GCC target ("avx2")
/// #pragma GCC optimize("unroll-loops")
 
        using namespace std;
 
        int timer;
        vector<vector<int>> adj;
        int tin[1001], out[1001];
        void dfs(int v, int p) {
            tin[v] = timer++;
            for(int u : adj[v]) {
                if(u != p) dfs(u, v);
            }
            out[v] = timer - 1;
        }
 
        vector<int> label(int N, int K, vector<int> U, vector<int> V) {
            vector<int> res(N, 0); timer = 0;
          	adj = vector<vector<int>>(N);
            for(int l = 0; l < N - 1; l++) {
                int X = U[l], Y = V[l];
                adj[X].push_back(Y), adj[Y].push_back(X);
            }
            dfs(0, 0);
            for(int l = 0; l < N; l++)
                res[l] = tin[l] * 1000 + out[l];
            return res;
        }
        int find_next_station(int S, int T, vector<int> R) {
          	vector<int> C = R;
          	sort(C.begin(), C.end());
          	int P;
            for(auto u : C) {
                int I = u / 1000, O = u % 1000;
                int X = T / 1000, Y = T % 1000;
              	if(S / 1000 >= I && S % 1000 <= O) {
                  P = u; continue;
                }
                if(X >= I && X <= O)
                    return u;
            }
          	return P;
        }

Compilation message (stderr)

stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:38:35: warning: unused variable 'Y' [-Wunused-variable]
   38 |                 int X = T / 1000, Y = T % 1000;
      |                                   ^
stations.cpp:35:16: warning: 'P' may be used uninitialized in this function [-Wmaybe-uninitialized]
   35 |            int P;
      |                ^

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