# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
984919 | 2024-05-17T08:21:57 Z | 54skyxenon | 악어의 지하 도시 (IOI11_crocodile) | C++17 | 0 ms | 0 KB |
// https://oj.uz/problem/view/IOI11_crocodile /** Needed for linking!!! */ #include "crocodile.h" #include <bits/stdc++.h> using namespace std; #define ll long long #define INF (ll)1e18 vector<map<int, int>> graph; vector<bool> is_exit; vector<bool> visited; int dfs(int curr) { if (is_exit[curr]) { return 0; } vector<ll> distances; for (auto [nei, weight] : graph[curr]) { if (!visited[nei]) { visited[nei] = true; distances.push_back(weight + dfs(nei)); } } sort(distances.begin(), distances.end()); if (distances.size() < 2) { throw runtime_error("Oops"); } return distances[1]; } int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]) { graph.resize(N); is_exit.resize(N); visited.resize(N); for (int i = 0; i < M; i++) { graph[R[i][0]][R[i][1]] = graph[R[i][1]][R[i][0]] = L[i]; } for (int i = 0; i < K; i++) { is_exit[P[i]] = true; } try { visited[0] = true; return dfs(0); } catch (...) { for (int nei : graph[0]) { assert(false); } return 0; } }