Submission #1073793

#TimeUsernameProblemLanguageResultExecution timeMemory
1073793thatsgonzalezStations (IOI20_stations)C++14
69.87 / 100
627 ms1700 KiB
#include "stations.h" #include <vector> #include <bits/stdc++.h> using namespace std; #define pb push_back int mn = 0; map<int,bool> mp; int dfs(int node, int p, vector<vector<int>> &g, vector<int> &sub_tree){ for(auto &x: g[node]){ if(x == p) continue; sub_tree[node]+=dfs(x,node,g,sub_tree); } return sub_tree[node]=sub_tree[node]+1; } void calc_labels(int node, int p, int t, vector<vector<int>> &g, vector<int> &sub_tree, vector<int> &labels){ if(t){ while(mp[mn]) mn++; mp[mn+sub_tree[node]-1] = 1; labels[node] = mn+sub_tree[node]-1; while(mp[mn]) mn++; } else{ while(mp[mn]) mn++; mp[mn] = 1; labels[node] = mn; while(mp[mn]) mn++; } for(auto &x: g[node]){ if(x == p) continue; int turno = t^1; calc_labels(x,node,turno,g,sub_tree,labels); } } std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) { vector<int> sub_tree(n,0); vector<vector<int>> g(n); vector<int> labels(n); for(int i = 0; i<n-1; i++){ g[u[i]].pb(v[i]); g[v[i]].pb(u[i]); } dfs(0,-1,g,sub_tree); calc_labels(0,-1,1,g,sub_tree,labels); return labels; } int find_next_station(int s, int t, std::vector<int> c) { if(c[0]>s){ int l = s; for(int i = 0; i<c.size()-1; i++){ if(t>=l and t<=c[i]){ return c[i]; } else l = c[i]+1; } return c.back(); } else{ int r = s; for(int i = c.size()-1; i; i--){ if(t>=c[i] and t<=r) return c[i]; else r = c[i]-1; } return c[0]; } }

Compilation message (stderr)

stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:70:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   70 |   for(int i = 0; i<c.size()-1; i++){
      |                  ~^~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...