Submission #1080819

#	Time	Username	Problem	Language	Result	Execution time	Memory
1080819		The_Blitz	Stations (IOI20_stations)	C++17	0 / 100	588 ms	940 KiB

stations

    // TODO -> PONER ALGO AQUI
    // amigos y no mas
    // I'm in the coolest driver high
    // += O(logn) ; + = O(n)
     
    #include <bits/stdc++.h>
    #include "stations.h"
    #include <variant>
    #define fastio ios_base::sync_with_stdio(0); cin.tie(0)
    #define ones(x) __builtin_popcount(x)
    #define loga2(x) __builtin_ctzll(x)
    #define pos2(x) __builtin_clzll(x)
     
    using namespace std;
     
    typedef long long ll;
    typedef pair<int,int> ii;
    typedef pair<int,ii> iii;
    typedef pair <ii, ii> iiii;
    typedef vector <int> vi;
    typedef vector <ii> vii;
     
    const int mod = 998244353;
    const ll inf = (1LL<<62)-1;
     
    struct HASH{
      size_t operator()(const pair<int,int>&x)const{
        return hash<long long>()(((long long)x.first)^(((long long)x.second)<<32));
      }
    };
     
    vector< vector<int> > G;
    vector<int> order;
    vector<bool> vis;
    int N;
    int cont = 0;
     
    void dfs(int x){
      for(int i=0; i<G[x].size(); i++){
        int &y = G[x][i];
        if(!vis[y]){
          vis[y] = true;
          dfs(y);
        }
      }
      order[x] = cont; cont++;
    }
     
     
    vector<int> label(int n, int k, vector<int> u, vector<int> v) {
      N = n;
      vector<int> labels(n);
     
      G.clear(); 
      order.clear(); 
      vis.clear();
     
      G.resize(n);
      order.resize(n);
      vis.resize(n,0);
     
      for(int i=0; i<n-1;i++){
        G[u[i]].push_back(v[i]);
        G[v[i]].push_back(u[i]);
      }
     
      vis[0] = true;
      dfs(0);
     
      for (int i = 0; i < n; i++) {
        labels[i] = order[i];
      }
     
      return labels;
    }
     
    int find_next_station(int s, int t, std::vector<int> c) {
     if(c.size() == 1) return c[0];
      if(s < t){
        auto it = *upper_bound(c.begin() , c.end() , s);
        return it;
      }
      // else t>s 
      auto it = lower_bound(c.begin() , c.end() , s); it--;
     
      return *it;
    }

Compilation message (stderr)

stations.cpp: In function 'void dfs(int)':
stations.cpp:39:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |       for(int i=0; i<G[x].size(); i++){
      |                    ~^~~~~~~~~~~~

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