# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
622616 | 2022-08-04T12:26:15 Z | BhavayGoyal | Crocodile's Underground City (IOI11_crocodile) | C++14 | 0 ms | 0 KB |
#include <bits/stdc++.h> using namespace std; #define vi vector<int> #define vii vector<vector<int>> #define pb push_back #define all(x) x.begin(), x.end() #define f first #define s second #define endl "\n" const int N = 1005; vector<array<int, 2>> g[N]; vi temp[N]; int dis[N], vis[N]; void dfs(int src) { vis[src] = true; for (auto ch : g[src]) { int child = ch[0], wt = ch[1]; if (!vis[child]) { dfs(child); temp[src].pb(dis[child]+wt); dis[src] = min(dis[src], dis[child] + wt); } } if (g[src].size() == 1) dis[src] = 0; else { sort (all(temp[src])); dis[src] = temp[src][1]; } } int travel_plan(int n, int m, int *r[2], int* l, int k, int *p) { memset(dis, 63, sizeof dis); vii edges; vi entry; for (int i = 0; i < m; i++) edges.pb({r[i][0], r[i][1]}); for (int i = 0; i < m; i++) edges[i].pb(l[i]); for (int i = 0; i < k; i++) { int x = p[i]; entry.pb(x); } for (auto i : edges) { int a = i[0], b = i[1], c = i[2]; g[a].pb({b, c}); g[b].pb({a, c}); } dfs(0); cout << dis[0]; }