Submission #806841

#	Time	Username	Problem	Language	Result	Execution time	Memory
806841		borisAngelov	Stations (IOI20_stations)	C++17	0 / 100	1565 ms	2097152 KiB

stations

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

using namespace std;

const int maxn = 1005;

vector<int> g[maxn];

int tim = 0;
int in[maxn];
int out[maxn];

void dfs(int node, int par)
{
    in[node] = tim++;

    for (int i = 0; i < g[node].size(); ++i)
    {
        if (g[node][i] != par)
        {
            dfs(g[node][i], node);
        }
    }

    out[node] = tim - 1;
}

vector<int> label(int n, int k, vector<int> u, vector<int> v)
{
    for (int i = 0; i < n - 1; ++i)
    {
        int x = u[i];
        int y = v[i];

        g[x].push_back(y);
        g[y].push_back(x);
    }

    dfs(0, -1);

    vector<int> hashed;

    for (int i = 0; i < n; ++i)
    {
        assert(0 <= in[i] && in[i] <= 999 && 0 <= out[i] && out[i] <= 999);
        hashed.push_back(in[i] * 1000 + out[i]);
    }

    return hashed;
}

int find_next_station(int s, int t, vector<int> c)
{
    int in_s = s / 1000;
    int out_s = s % 1000;

    int in_t = t / 1000;
    int out_t = t % 1000;

    if (in_s <= in_t && out_t <= out_s)
    {
        for (int i = 0; i < c.size(); ++i)
        {
            int in_i = c[i] / 1000;
            int out_i = c[i] % 1000;

            if ((in_s <= in_i && out_i <= out_s) && (in_i <= in_t && out_t <= out_i))
            {
                return c[i];
            }
        }
    }
    else
    {
        for (int i = 0; i < c.size(); ++i)
        {
            int in_i = c[i] / 1000;
            int out_i = c[i] % 1000;

            if (in_i <= in_s && out_s <= out_i)
            {
                return c[i];
            }
        }
    }

    return -1;
}

Compilation message (stderr)

stations.cpp: In function 'void dfs(int, int)':
stations.cpp:18:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   18 |     for (int i = 0; i < g[node].size(); ++i)
      |                     ~~^~~~~~~~~~~~~~~~
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:63:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |         for (int i = 0; i < c.size(); ++i)
      |                         ~~^~~~~~~~~~
stations.cpp:76:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   76 |         for (int i = 0; i < c.size(); ++i)
      |                         ~~^~~~~~~~~~

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