49162023-04-07 00:19:53TomaSajtA lehető legkevesebb metróval utazás (40 pont)cpp17Accepted 40/40114ms25600 KiB
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() {
  cin.tie(0), ios::sync_with_stdio(0);
  int n, m, s, e;
  cin >> n >> m >> s >> e;
  vector<vector<int>> station_to_lines(m + 1);
  vector<vector<int>> line_to_stations(n + 1);
  for (int i = 1; i <= n; i++) {
    int a;
    cin >> a;
    while (a--) {
      int b;
      cin >> b;
      line_to_stations[i].push_back(b);
      station_to_lines[b].push_back(i);
    }
  }
  vector<bool> used_line(n + 1);
  vector<array<int, 2>> prev(m + 1);
  vector<int> dist(m + 1, -1);
  prev[s] = {-1, -1}; // {station,line}
  dist[s] = 0;
  queue<int> q;
  q.push(s);
  while (!q.empty()) {
    int curr_station = q.front();
    q.pop();
    for (int &line : station_to_lines[curr_station]) {
      if (used_line[line])
        continue;
      used_line[line] = 1;
      for (int &next_station : line_to_stations[line]) {
        if (dist[next_station] != -1)
          continue;
        dist[next_station] = dist[curr_station] + 1;
        prev[next_station] = {curr_station, line};
        q.push(next_station);
      }
    }
  }
  cout << dist[e] << endl;
  if (dist[e] == -1)
    return 0;

  stack<int> res;
  array<int, 2> currState = prev[e];
  while (currState[0] != -1) {
    res.push(currState[1]);
    currState = prev[currState[0]];
  }
  while (!res.empty()) {
    cout << res.top() << ' ';
    res.pop();
  }
}
SubtaskSumTestVerdictTimeMemory
base40/40
1Accepted0/03ms1828 KiB
2Accepted0/04ms3392 KiB
3Accepted2/23ms2264 KiB
4Accepted2/23ms2480 KiB
5Accepted2/23ms2916 KiB
6Accepted2/23ms2964 KiB
7Accepted2/23ms3444 KiB
8Accepted2/23ms3680 KiB
9Accepted2/24ms4032 KiB
10Accepted2/24ms3964 KiB
11Accepted2/23ms3760 KiB
12Accepted2/24ms4672 KiB
13Accepted2/24ms4816 KiB
14Accepted2/24ms4720 KiB
15Accepted2/2114ms25264 KiB
16Accepted2/2111ms25464 KiB
17Accepted2/2114ms25348 KiB
18Accepted2/2111ms25600 KiB
19Accepted2/24ms4512 KiB
20Accepted2/24ms4764 KiB
21Accepted2/24ms4384 KiB
22Accepted2/24ms5236 KiB