Submission #683690

#	Time	Username	Problem	Language	Result	Execution time	Memory
683690		sharaelong	Stations (IOI20_stations)	C++17	52.32 / 100	1049 ms	760 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> pii;

const int MAX_N = 1000;
vector<int> adj[MAX_N];
int tour_cnt = 0;
pii tour[MAX_N];

void dfs(int here, int parent = -1) {
    tour[here].first = tour_cnt++;
    for (int there: adj[here]) {
        if (there != parent) {
            dfs(there, here);
        }
    }
    tour[here].second = tour_cnt-1;
}

vector<int> label(int n, int k, vector<int> u, vector<int> v) {
    for (int i=0; i<n; ++i) adj[i].clear();
    tour_cnt = 0;
    
    for (int i=0; i<n-1; ++i) {
        adj[u[i]].push_back(v[i]);
        adj[v[i]].push_back(u[i]);
    }
    dfs(0);
    assert(tour_cnt == n);

    const int MAX_N = 1000;
	vector<int> labels(n);
	for (int i=0; i<n; i++) labels[i] = tour[i].first * MAX_N + tour[i].second;
    // for (int i=0; i<n; ++i) cout << tour[i].first << ' ' << tour[i].second << endl;
	return labels;
}

int find_next_station(int s, int t, vector<int> c) {
    const int MAX_N = 1000;
    
    for (int i=0; i<c.size(); ++i) {
        if (c[i] < s) {
            assert(i == 0);
            continue;
        }
        int x = c[i] / MAX_N, y = c[i] % MAX_N;
        if (x <= t / MAX_N && t % MAX_N <= y) return c[i];
    }
    return c[0];
}

Compilation message (stderr)

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

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