#include "dreaming.h"
#include <bits/stdc++.h>
using namespace std;
using pii = pair <int, int>;
using vi = vector <int>;
#define F first
#define S second
#define mp make_pair
#define pb push_back
#define N 100005
vector <pii> edg[N];
vector <int> se;
int dist, dp[N], dp1[N], dp2[N], up[N], up1[N], ans;
int vis[N];
void dfs(int u, int mx){
// cerr << u<< endl;
vis[u] = 1;
if (up[u] < mx){
up1[u] = up[u];
up[u] = mx;
}
else if (up1[u] < mx) up1[u] = mx;
for (pii v : edg[u])
if (!vis[v.F]){
dfs(v.F, max(mx, dp[u]) + v.S);
if (dp[u] < dp[v.F] + v.S) {
dp1[u] = dp[u];
dp[u] = dp[v.F] + v.S;
}
else if (dp1[u] < dp[v.F] + v.S) dp1[u] = dp[v.F] + v.S;
}
}
void dfs1(int u, int mx){
vis[u] = 2;
if (up[u] < mx){
up1[u] = up[u];
up[u] = mx;
}
else if (up1[u] < mx) up1[u] = mx;
for (int i = edg[u].size() - 1;i >= 0;i--){
pii v = edg[u][i];
if (vis[v.F] < 2){
dfs1(v.F, max(mx, dp2[u]) + v.S);
dp2[u] = max(dp2[u], dp2[v.F] + v.S);
}
}
ans = max({ans, dp[u] + dp1[u], up[u] + up1[u], dp[u] + up[u]});
dist = min(dist, max({dp[u], dp1[u], up[u], up1[u]}));
}
int travelTime(int n, int m, int L, int A[], int B[], int T[]) {
// return 0;
for (int i = 0;i < m;i++){
edg[A[i]].pb(mp(B[i], T[i]));
edg[B[i]].pb(mp(A[i], T[i]));
}
int k = 0;
for (int i = 0;i < n;i++)
if (!vis[i]){
k++;
dist = 1e9;
// cerr << i<< endl;
dfs(i, 0);
// return 0;
dfs1(i, 0);
// cerr << dist << endl;
se.pb(dist);
}
sort(se.begin(), se.end());
if (se.size() == 1) return ans;
if (se.size() == 2) return max(ans, se[0] + se[1] + L);
return max({se.back() + se[k - 2] + L, se[k - 2] + se[k - 3] + 2 * L, ans});
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
88 ms |
12792 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
88 ms |
12792 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
88 ms |
12792 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
29 ms |
7544 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
88 ms |
12792 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
88 ms |
12792 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |