Submission #730093

#	Time	Username	Problem	Language	Result	Execution time	Memory
730093		danikoynov	Stations (IOI20_stations)	C++14	100 / 100	1012 ms	892 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>
using namespace std;

const int maxn = 1010;
vector < int > g[maxn], order, my_labels;
int in[maxn], out[maxn], timer, par[maxn], depth[maxn];
void dfs(int v, int p)
{
    if (depth[v] % 2 == 0)
    in[v] = timer ++;
    par[v] = p;
    order.push_back(v);
    for (int u : g[v])
    {

        if (u == p)
            continue;
        depth[u] = depth[v] + 1;
        dfs(u, v);
    }
    if (depth[v] % 2 == 1)
    out[v] = timer ++;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v)
{
    for (int i = 0; i < n; i ++)
    {
        g[i].clear();
    }
    timer = 0;
    order.clear();
    my_labels.clear();
    for (int i = 0; i < v.size(); i ++)
    {
        g[u[i]].push_back(v[i]);
        g[v[i]].push_back(u[i]);
    }
    my_labels.resize(n);
    dfs(0, 0);
    for (int i = 0; i < n; i ++)
    {
        if (depth[i] % 2 == 0)
        my_labels[i] = in[i];
        else
            my_labels[i] = out[i];
    }


    return my_labels;
}

int used[maxn];
int find_next_station(int s, int t, vector<int> c)
{
    ///return 0;
    sort(c.begin(), c.end());
    if (c.size() == 1)
        return c[0];
    int m = c.size();
    if (s < c[0])
    {
        int tin = s;
        int tout = c[m - 2] + 1;
        if (t < tin || t > tout)
        return c[m - 1];
        int pt = 0;
        while(c[pt] < t)
            pt ++;
        return c[pt];
    }
    else
    {
        int tin = c[1] - 1;
        int tout = s;
        if (t < tin || t > tout)
        return c[0];
        int pt = c.size() - 1;
        while(c[pt] > t)
            pt --;
        return c[pt];
    }


}

Compilation message (stderr)

stations.cpp: In function 'std::vector<int> label(int, int, std::vector<int>, std::vector<int>)':
stations.cpp:34:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |     for (int i = 0; i < v.size(); i ++)
      |                     ~~^~~~~~~~~~

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