#include "crocodile.h"
#include <vector>
#include <algorithm>
using namespace std;
#define ll long long
#define pb push_back
#define pii pair<ll, ll>
#define f first;
#define s second;
#define oo 1000000000000000000ll
vector<vector<pii>> ed;
vector<bool> ext;
vector<int> visited;
vector<ll> res;
int dfs(int v){
visited[v] = 1;
vector<int> cur;
for (auto [x, l] : ed[v]){
if (visited[x] == 1) continue;
if (ext[x]) cur.pb(l);
else if (visited[x] == 2) cur.pb(res[x] + l);
else cur.pb(dfs(x) + l);
}
visited[v] = 2;
sort(cur.begin(), cur.end());
if (cur.size() <= 1) return res[v] = oo;
return res[v] = cur[1];
}
int travel_plan(int n, int m, int r[][2], int l[], int k, int p[]){
ed.resize(n);
ext.resize(n);
visited.resize(n);
res.resize(n);
for (int i = 0; i < m; ++i) {
ed[r[i][0]].pb({r[i][1], l[i]});
ed[r[i][1]].pb({r[i][0], l[i]});
}
for (int i = 0; i < k; ++i) {
ext[p[i]] = true;
}
int ans = dfs(0);
return ans;
}
Compilation message
crocodile.cpp: In function 'int dfs(int)':
crocodile.cpp:22:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
22 | for (auto [x, l] : ed[v]){
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
0 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
332 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
1 ms |
332 KB |
Output is correct |
7 |
Correct |
1 ms |
460 KB |
Output is correct |
8 |
Correct |
1 ms |
460 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
0 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
332 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
1 ms |
332 KB |
Output is correct |
7 |
Correct |
1 ms |
460 KB |
Output is correct |
8 |
Correct |
1 ms |
460 KB |
Output is correct |
9 |
Correct |
2 ms |
716 KB |
Output is correct |
10 |
Incorrect |
1 ms |
332 KB |
Output isn't correct |
11 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
0 ms |
204 KB |
Output is correct |
3 |
Correct |
1 ms |
332 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Correct |
1 ms |
332 KB |
Output is correct |
7 |
Correct |
1 ms |
460 KB |
Output is correct |
8 |
Correct |
1 ms |
460 KB |
Output is correct |
9 |
Correct |
2 ms |
716 KB |
Output is correct |
10 |
Incorrect |
1 ms |
332 KB |
Output isn't correct |
11 |
Halted |
0 ms |
0 KB |
- |