This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
vector <vector <pair <int, int>>> adj;
vector <bool> is_exit;
ll dfs(int s, int p) {
if (adj[s].size() == 1 && s != p)
return 0;
vector <ll> x;
for (auto i : adj[s]) {
if (i.first == p)
continue;
x.push_back(dfs(i.first, s) + i.second);
}
sort(x.begin(), x.end());
return x[1];
}
int travel_plan(int N, int M, int R[][2], int W[], int K, int P[]) {
adj.resize(N+1);
is_exit = vector <bool> (N+1);
for (int i = 0; i < M; i++) {
int a = R[i][0], b = R[i][1];
int w = W[i];
adj[a].push_back({b, w});
adj[b].push_back({a, w});
}
for (int i = 0; i < K; i++)
is_exit[P[i]] = true;
if (M > N-1)
return 0;
return dfs(0, 0);
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |