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>
#include "dreaming.h"
using namespace std;
using ll = long long;
int travelTime(int n, int m, int l, int a[], int b[], int t[]) {
vector < vector < pair < int, ll > > > adj (n);
for (int i = 0; i < m; ++i)
{
adj[a[i]].push_back ({ b[i], (ll) t[i] });
adj[b[i]].push_back ({ a[i], (ll) t[i] });
}
vector < bool > vi (n);
vector < tuple < ll, ll, ll > > dd (n);
auto dfs1 = [&] (auto self, int v, int p) -> ll
{
vi[v] = true;
auto &[x, m, y] = dd[v];
m = -1;
for (auto &c : adj[v])
if (c.first != p)
{
ll z = self (self, c.first, v) + c.second;
if (z > x) y = x, x = z, m = c.first;
else if (z > y) y = z;
}
return x;
};
auto dfs2 = [&] (auto self, int v, int p, ll z) -> pair < ll, ll >
{
auto &[x, m, y] = dd[v];
if (z > x) y = x, x = z, m = p;
else if (z > y) y = z;
pair < ll, ll > ans = { x, v };
for (auto &c : adj[v])
if (c.first != p)
ans = min (ans, self (self, c.first, v, (c.first == m ? y : x) + c.second));
return ans;
};
vector < int > v;
for (int i = 0; i < n; ++i)
if (!vi[i])
dfs1 (dfs1, i, -1), v.push_back (dfs2 (dfs2, i, -1, 0).second);
ll ix = 0, iy = 0;
for (auto &k : v)
{
auto &[x, m, y] = dd[k];
auto jx = x, jy = y;
if (ix + l > jx) jy = jx, jx = ix + l;
else if (ix + l > jy) jy = ix + l;
if (iy + l > jy) jy = iy + l;
if (x + l > ix) iy = ix, ix = x + l;
else if (x + l > iy) iy = x + l;
if (y + l > iy) iy = y + l;
if (jx < ix) ix = jx, iy = jy;
}
return ix + iy;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |