Submission #586616

#	Time	Username	Problem	Language	Result	Execution time	Memory
586616		PiejanVDC	Stations (IOI20_stations)	C++17	0 / 100	938 ms	672 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>

using namespace std;

vector<int>adj[1000];
vector<int>labelss;
int id;

int dfs(int u, int e = -1) {
    labelss[u] = 1000*id;
    id++;
    int sz = 0;
    for(auto z : adj[u]) if(z != e) {
        sz += dfs(z, u);
    }
    labelss[u] += sz;
    return sz+1;
}

std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
    for(int i = 0 ; i < n ; i++)
        adj[i].clear();
    labelss.clear();
    id = 0;
    for(int i = 0 ; i < n-1 ; i++)
        adj[u[i]].push_back(v[i]), adj[v[i]].push_back(u[i]);
    labelss.resize(n);
    dfs(0);
    return labelss;
}

int find_next_station(int s, int t, std::vector<int> c) {
    int ID = s / 1000;
    int SZ = s % 1000;
    int ID_T = t / 1000;

    if(ID_T < ID || ID_T > ID + SZ) {
        int l, mn = INT_MAX;
        for(int i = 0 ; i < c.size() ; i++) {
            if(c[i] / 1000 < mn) {
                mn = c[i] / 1000;
                l = i;
            }
        }
        return c[l];
    }

    int curr = ID;
    for(int i = 1 ; i < c.size() ; i++) {
        curr += 1 + (c[i] % 1000);
        if(curr >= ID_T)
            return c[i];
    }
    return -1;
}

Compilation message (stderr)

stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:40:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |         for(int i = 0 ; i < c.size() ; i++) {
      |                         ~~^~~~~~~~~~
stations.cpp:50:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   50 |     for(int i = 1 ; i < c.size() ; i++) {
      |                     ~~^~~~~~~~~~
stations.cpp:46:19: warning: 'l' may be used uninitialized in this function [-Wmaybe-uninitialized]
   46 |         return c[l];
      |                   ^

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