제출 #985465

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
985465		Seb	Stations (IOI20_stations)	C++17	100 / 100	643 ms	1644 KiB

stations

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

using namespace std;

void dfs(int nodo, int p, vector<vector<int>> &g, vector <int> &labels, int caso, int &t) {
    if (caso == 0) labels[nodo] = t++;
    for (auto &it : g[nodo]) {
        if (it == p) continue;
        dfs(it, nodo, g, labels, (caso^1), t);
    }
    if (caso == 1) labels[nodo] = t++;
    return;
}

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

int find_next_station(int s, int t, std::vector<int> c) {
    if (c.size() == 1) return c[0];
    if (c[0] > s) {
        if (t < s || t > c[c.size() - 2]) return c[c.size() - 1];
        for (int i = c.size() - 1; i >= 0; i--)
            if (c[i] < t) return c[i + 1];
        return c[0];
    }
    else {
        if (t > s || t < c[1]) return c[0];
        for (int i = 0; i < c.size(); i++)
            if (c[i] > t) return c[i - 1];
        return c[c.size() - 1];
    }
	return c[0];
}

컴파일 시 표준 에러 (stderr) 메시지

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

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