Submission #604699

#	Time	Username	Problem	Language	Result	Execution time	Memory
604699		SamAnd	Stations (IOI20_stations)	C++17	Compilation error	0 ms	0 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
#define m_p make_pair
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
typedef long long ll;
const int N = 1003;

vector<int> g[N];

int ti;
int tin[N], tout[N];
int d[N];
void dfs(int x, int p)
{
    if (x == p)
        d[x] = 0;
    else
        d[x] = d[p] + 1;
    tin[x] = ti++;
    for (int i = 0; i < g[x].size(); ++i)
    {
        int h = g[x][i];
        if (h == p)
            continue;
        dfs(h, x);
    }
    tout[x] = ti++;
}

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

    ti = 0;
    dfs(0, 0);

    vector<int> ans;
    for (int x = 0; x < n; ++x)
    {
        if (d[x] % 2 == 0)
            ans.push_back(tin[x]);
        else
            ans.push_back(tout[x]);
    }

    vector<int> v = ans;
    sort(all(v));
    for (int i = 0; i < sz(ans); ++i)
        ans[i] = lower_bound(all(v), ans[i]) - v.begin();
    return ans;
}

int find_next_station(int x, int y, std::vector<int> v)
{
    for (int i = 0; i < sz(v); ++i)
    {
        if (v[i] == y)
            return y;
    }

    if (x == 0)
    {
        sort(all(v));
        for (int i = 0; i < sz(v) - 1; ++i)
        {
            if (v[i] < y && y < v[i + 1])
                return v[i + 1];
        }
        return v[0];
    }

    if (x > v[0])
    {
        sort(all(v));
        for (int i = 1; i < sz(v) - 1; ++i)
        {
            if (v[i] < y && y < v[i + 1])
                return v[i];
        }
        if (v.back() < y && y < x)
            return v.back();
        return v[0];
    }
    else
    {
        sort(all(v));
        for (int i = 0; i < sz(v) - 2; ++i)
        {
            if (v[i] < y && y < v[i + 1])
                return v[i + 1];
        }
        if (x < y && y < v[0])
            return v[0];
        return v.back();
    }
}

/*
1
5 2000
0 1
0 2
2 3
2 4
1
1 4 -1
*/

Compilation message (stderr)

stations.cpp: In function 'void dfs(int, int)':
stations.cpp:24:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   24 |     for (int i = 0; i < g[x].size(); ++i)
      |                     ~~^~~~~~~~~~~~~
stations.cpp: In function 'std::vector<int> label(int, int, std::vector<int>, std::vector<int>)':
stations.cpp:55:17: error: declaration of 'std::vector<int> v' shadows a parameter
   55 |     vector<int> v = ans;
      |                 ^
stations.cpp:34:75: note: 'std::vector<int> v' previously declared here
   34 | std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
      |                                                          ~~~~~~~~~~~~~~~~~^