| 16064 | 2025-03-30 11:09:01 | szil | Autó-tortúra | cpp17 | Accepted 100/100 | 801ms | 71344 KiB |
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int MAXN = 3001;
const int MAXB = 151;
const ll INF = 1e14;
vector<int> g[MAXN];
ll bdist[MAXB][MAXB], dist[MAXN][MAXN];
int n, m, k;
void bfs(int st) {
fill(dist[st], dist[st]+MAXN, INF);
queue<int> q;
q.push(st);
dist[st][st] = 0;
while (!q.empty()) {
int u = q.front(); q.pop();
for (int v : g[u]) {
if (dist[st][v] > dist[st][u] + 1) {
dist[st][v] = dist[st][u] + 1;
q.push(v);
}
}
}
}
void solve() {
cin >> n >> m >> k;
for (int i = 0; i < m; i++) {
int a, b; cin >> a >> b;
g[a].emplace_back(b);
g[b].emplace_back(a);
}
int l; cin >> l;
vector<int> terv(l+1);
for (int i = 1; i <= l; i++) cin >> terv[i];
int b; cin >> b;
vector<int> ben(b+1);
for (int i = 1; i <= b; i++) cin >> ben[i];
for (int i = 1; i <= n; i++) {
bfs(i);
}
for (int i = 1; i <= b; i++) {
for (int j = 1; j <= b; j++) {
bdist[i][j] = INF;
}
bdist[i][i] = 0;
}
for (int i = 1; i <= b; i++) {
for (int j = 1; j <= b; j++) {
int u = ben[i];
int v = ben[j];
if (dist[u][v] <= k) {
bdist[i][j] = min(bdist[i][j], dist[u][v]);
}
}
}
for (int c = 1; c <= b; c++) {
for (int i = 1; i <= b; i++) {
for (int j = 1; j <= b; j++) {
bdist[i][j] = min(bdist[i][j], bdist[i][c] + bdist[c][j]);
}
}
}
vector<ll> dp_last(k+1);
for (int a = 2; a <= l; a++) {
vector<ll> dp_curr(k+1, INF);
int start = terv[a-1];
int goal = terv[a];
for (int i = 1; i <= b; i++) {
for (int j = 1; j <= b; j++) {
ll d = dist[start][ben[i]] + bdist[i][j] + dist[ben[j]][goal];
ll need = dist[start][ben[i]];
ll left = k - dist[ben[j]][goal];
if (need > k) continue;
if (left < 0) continue;
d += dp_last[need];
dp_curr[left] = min(dp_curr[left], d);
}
}
for (int i = 0; i <= k - dist[start][goal]; i++) {
dp_curr[i] = min(dp_curr[i], dist[start][goal] + dp_last[i + dist[start][goal]]);
}
for (int i = k-1; i >= 0; i--) {
dp_curr[i] = min(dp_curr[i], dp_curr[i+1]);
}
dp_last = dp_curr;
}
ll ans = *min_element(dp_last.begin(), dp_last.end());
cout << (ans >= INF ? -1 : ans) << "\n";
}
int main() {
ios::sync_with_stdio(0); cin.tie(0);
int t = 1;
// cin >> t;
while (t--) solve();
return 0;
}| Subtask | Sum | Test | Verdict | Time | Memory | ||
|---|---|---|---|---|---|---|---|
| subtask1 | 0/0 | ||||||
| 1 | Accepted | 1ms | 748 KiB | ||||
| 2 | Accepted | 412ms | 52824 KiB | ||||
| subtask2 | 12/12 | ||||||
| 3 | Accepted | 1ms | 564 KiB | ||||
| 4 | Accepted | 1ms | 564 KiB | ||||
| 5 | Accepted | 1ms | 756 KiB | ||||
| 6 | Accepted | 1ms | 564 KiB | ||||
| 7 | Accepted | 1ms | 492 KiB | ||||
| 8 | Accepted | 1ms | 564 KiB | ||||
| 9 | Accepted | 1ms | 580 KiB | ||||
| 10 | Accepted | 1ms | 564 KiB | ||||
| subtask3 | 18/18 | ||||||
| 11 | Accepted | 1ms | 564 KiB | ||||
| 12 | Accepted | 1ms | 564 KiB | ||||
| 13 | Accepted | 1ms | 756 KiB | ||||
| 14 | Accepted | 1ms | 564 KiB | ||||
| 15 | Accepted | 1ms | 492 KiB | ||||
| 16 | Accepted | 1ms | 564 KiB | ||||
| 17 | Accepted | 1ms | 580 KiB | ||||
| 18 | Accepted | 1ms | 564 KiB | ||||
| 19 | Accepted | 4ms | 2872 KiB | ||||
| 20 | Accepted | 19ms | 6512 KiB | ||||
| 21 | Accepted | 18ms | 6316 KiB | ||||
| 22 | Accepted | 17ms | 6520 KiB | ||||
| 23 | Accepted | 8ms | 6392 KiB | ||||
| 24 | Accepted | 8ms | 6516 KiB | ||||
| 25 | Accepted | 20ms | 6504 KiB | ||||
| 26 | Accepted | 18ms | 6476 KiB | ||||
| 27 | Accepted | 10ms | 6488 KiB | ||||
| subtask4 | 25/25 | ||||||
| 28 | Accepted | 1ms | 564 KiB | ||||
| 29 | Accepted | 1ms | 564 KiB | ||||
| 30 | Accepted | 1ms | 756 KiB | ||||
| 31 | Accepted | 1ms | 564 KiB | ||||
| 32 | Accepted | 1ms | 492 KiB | ||||
| 33 | Accepted | 1ms | 564 KiB | ||||
| 34 | Accepted | 1ms | 580 KiB | ||||
| 35 | Accepted | 1ms | 564 KiB | ||||
| 36 | Accepted | 488ms | 71080 KiB | ||||
| 37 | Accepted | 425ms | 71220 KiB | ||||
| 38 | Accepted | 467ms | 71180 KiB | ||||
| 39 | Accepted | 477ms | 71072 KiB | ||||
| 40 | Accepted | 428ms | 70964 KiB | ||||
| 41 | Accepted | 407ms | 70956 KiB | ||||
| 42 | Accepted | 425ms | 71220 KiB | ||||
| 43 | Accepted | 293ms | 71220 KiB | ||||
| 44 | Accepted | 293ms | 71220 KiB | ||||
| 45 | Accepted | 149ms | 71224 KiB | ||||
| subtask5 | 24/24 | ||||||
| 46 | Accepted | 474ms | 70964 KiB | ||||
| 47 | Accepted | 437ms | 70964 KiB | ||||
| 48 | Accepted | 467ms | 71124 KiB | ||||
| 49 | Accepted | 419ms | 70888 KiB | ||||
| 50 | Accepted | 294ms | 70932 KiB | ||||
| 51 | Accepted | 483ms | 70924 KiB | ||||
| 52 | Accepted | 421ms | 70960 KiB | ||||
| 53 | Accepted | 463ms | 70964 KiB | ||||
| 54 | Accepted | 418ms | 70964 KiB | ||||
| 55 | Accepted | 282ms | 70964 KiB | ||||
| subtask6 | 21/21 | ||||||
| 56 | Accepted | 1ms | 564 KiB | ||||
| 57 | Accepted | 1ms | 564 KiB | ||||
| 58 | Accepted | 1ms | 756 KiB | ||||
| 59 | Accepted | 1ms | 564 KiB | ||||
| 60 | Accepted | 1ms | 492 KiB | ||||
| 61 | Accepted | 1ms | 564 KiB | ||||
| 62 | Accepted | 1ms | 580 KiB | ||||
| 63 | Accepted | 1ms | 564 KiB | ||||
| 64 | Accepted | 4ms | 2872 KiB | ||||
| 65 | Accepted | 19ms | 6512 KiB | ||||
| 66 | Accepted | 18ms | 6316 KiB | ||||
| 67 | Accepted | 17ms | 6520 KiB | ||||
| 68 | Accepted | 8ms | 6392 KiB | ||||
| 69 | Accepted | 8ms | 6516 KiB | ||||
| 70 | Accepted | 20ms | 6504 KiB | ||||
| 71 | Accepted | 18ms | 6476 KiB | ||||
| 72 | Accepted | 10ms | 6488 KiB | ||||
| 73 | Accepted | 488ms | 71080 KiB | ||||
| 74 | Accepted | 425ms | 71220 KiB | ||||
| 75 | Accepted | 467ms | 71180 KiB | ||||
| 76 | Accepted | 477ms | 71072 KiB | ||||
| 77 | Accepted | 428ms | 70964 KiB | ||||
| 78 | Accepted | 407ms | 70956 KiB | ||||
| 79 | Accepted | 425ms | 71220 KiB | ||||
| 80 | Accepted | 293ms | 71220 KiB | ||||
| 81 | Accepted | 293ms | 71220 KiB | ||||
| 82 | Accepted | 149ms | 71224 KiB | ||||
| 83 | Accepted | 474ms | 70964 KiB | ||||
| 84 | Accepted | 437ms | 70964 KiB | ||||
| 85 | Accepted | 467ms | 71124 KiB | ||||
| 86 | Accepted | 419ms | 70888 KiB | ||||
| 87 | Accepted | 294ms | 70932 KiB | ||||
| 88 | Accepted | 483ms | 70924 KiB | ||||
| 89 | Accepted | 421ms | 70960 KiB | ||||
| 90 | Accepted | 463ms | 70964 KiB | ||||
| 91 | Accepted | 418ms | 70964 KiB | ||||
| 92 | Accepted | 282ms | 70964 KiB | ||||
| 93 | Accepted | 365ms | 71216 KiB | ||||
| 94 | Accepted | 349ms | 71196 KiB | ||||
| 95 | Accepted | 444ms | 71216 KiB | ||||
| 96 | Accepted | 526ms | 71232 KiB | ||||
| 97 | Accepted | 170ms | 71224 KiB | ||||
| 98 | Accepted | 127ms | 24128 KiB | ||||
| 99 | Accepted | 486ms | 71220 KiB | ||||
| 100 | Accepted | 777ms | 71304 KiB | ||||
| 101 | Accepted | 800ms | 71292 KiB | ||||
| 102 | Accepted | 740ms | 71220 KiB | ||||
| 103 | Accepted | 629ms | 71220 KiB | ||||
| 104 | Accepted | 509ms | 71160 KiB | ||||
| 105 | Accepted | 619ms | 71092 KiB | ||||
| 106 | Accepted | 699ms | 71224 KiB | ||||
| 107 | Accepted | 570ms | 71220 KiB | ||||
| 108 | Accepted | 801ms | 71344 KiB | ||||
| 109 | Accepted | 170ms | 70968 KiB | ||||
| 110 | Accepted | 423ms | 70964 KiB | ||||
| 111 | Accepted | 432ms | 70960 KiB | ||||
| 112 | Accepted | 330ms | 70964 KiB | ||||
| 113 | Accepted | 368ms | 70964 KiB | ||||
| 114 | Accepted | 665ms | 71220 KiB | ||||
| 115 | Accepted | 467ms | 71220 KiB | ||||
| 116 | Accepted | 483ms | 71220 KiB | ||||
| 117 | Accepted | 722ms | 71220 KiB | ||||
| 118 | Accepted | 316ms | 71220 KiB | ||||