#include "dreaming.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MX = 1e5 + 5;
int N, M, L;
vector<pair<int,int>> g[MX];
bool vis[MX];
int cur = 0, mx = 0, z = 0, len = 0;
vector<int> vec;
int p = 0, q = 0;
pair<int,int> par[MX];
void dfs(int v) {
vis[v] = 1;
vec.push_back(v);
if(cur > mx) {
mx = cur;
z = v;
}
for(auto [u, w] : g[v]) {
if(!vis[u]) {
if(cur + w > len / 2 && !p && len > 0) {
p = cur;
q = len - cur;
}
cur += w;
par[u] = {v, w};
dfs(u);
cur -= w;
}
}
}
vector<pair<int,int>> P;
int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
for(int i = 0; i < M; i++) {
g[A[i]].push_back({B[i], T[i]});
g[B[i]].push_back({A[i], T[i]});
}
for(int i = 1; i <= N; i++) {
if(vis[i]) continue;
vec.clear();
cur = 0, mx = 0, z = 0, len = 0;
dfs(i);
for(auto x : vec) {
vis[x] = 0;
}
if(vec.size() == 1) {
for(auto x : vec) {
vis[x] = 1;
}
P.push_back({0, 0});
continue;
}
if(vec.size() == 2) {
for(auto x : vec) {
vis[x] = 1;
}
P.push_back({mx, 0});
continue;
}
vec.clear();
int tz = z;
cur = 0, mx = 0, z = 0;
dfs(tz);
len = mx;
int k = z, sum = 0;
while(k != tz) {
if(sum + par[k].second > len / 2 && sum <= len / 2) {
p = sum;
q = len - sum;
}
sum += par[k].second;
k = par[k].first;
}
vec.clear();
P.push_back({max(p, q), min(p, q)});
}
sort(P.rbegin(), P.rend());
int ans = 0;
for(auto [x, y] : P) {
ans = max(ans, x + y);
}
for(int i = 1; i < P.size(); i++) {
auto [x, y] = P[i];
ans = max(ans, P[0].first + L + x);
}
if(P.size() > 2) {
ans = max(ans, P[1].first + 2 * L + P[2].first);
}
return ans;
}
Compilation message
dreaming.cpp: In function 'int travelTime(int, int, int, int*, int*, int*)':
dreaming.cpp:101:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
101 | for(int i = 1; i < P.size(); i++) {
| ~~^~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
30 ms |
15056 KB |
Output is correct |
2 |
Correct |
36 ms |
14956 KB |
Output is correct |
3 |
Correct |
21 ms |
11736 KB |
Output is correct |
4 |
Correct |
5 ms |
6236 KB |
Output is correct |
5 |
Correct |
4 ms |
5464 KB |
Output is correct |
6 |
Correct |
8 ms |
7004 KB |
Output is correct |
7 |
Incorrect |
1 ms |
4700 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
4444 KB |
Output is correct |
2 |
Correct |
1 ms |
4444 KB |
Output is correct |
3 |
Incorrect |
1 ms |
4444 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
30 ms |
15056 KB |
Output is correct |
2 |
Correct |
36 ms |
14956 KB |
Output is correct |
3 |
Correct |
21 ms |
11736 KB |
Output is correct |
4 |
Correct |
5 ms |
6236 KB |
Output is correct |
5 |
Correct |
4 ms |
5464 KB |
Output is correct |
6 |
Correct |
8 ms |
7004 KB |
Output is correct |
7 |
Incorrect |
1 ms |
4700 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
17 ms |
7648 KB |
Output is correct |
2 |
Correct |
14 ms |
7644 KB |
Output is correct |
3 |
Correct |
16 ms |
7636 KB |
Output is correct |
4 |
Correct |
14 ms |
7644 KB |
Output is correct |
5 |
Correct |
15 ms |
7600 KB |
Output is correct |
6 |
Correct |
15 ms |
8412 KB |
Output is correct |
7 |
Correct |
14 ms |
7904 KB |
Output is correct |
8 |
Correct |
14 ms |
7608 KB |
Output is correct |
9 |
Correct |
14 ms |
7620 KB |
Output is correct |
10 |
Correct |
14 ms |
7892 KB |
Output is correct |
11 |
Correct |
1 ms |
4696 KB |
Output is correct |
12 |
Correct |
4 ms |
5844 KB |
Output is correct |
13 |
Correct |
4 ms |
5844 KB |
Output is correct |
14 |
Correct |
4 ms |
5844 KB |
Output is correct |
15 |
Correct |
4 ms |
5844 KB |
Output is correct |
16 |
Correct |
4 ms |
5844 KB |
Output is correct |
17 |
Correct |
4 ms |
5844 KB |
Output is correct |
18 |
Correct |
5 ms |
5676 KB |
Output is correct |
19 |
Correct |
4 ms |
5844 KB |
Output is correct |
20 |
Correct |
1 ms |
4444 KB |
Output is correct |
21 |
Correct |
1 ms |
4444 KB |
Output is correct |
22 |
Correct |
1 ms |
4952 KB |
Output is correct |
23 |
Correct |
4 ms |
5844 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
4444 KB |
Output is correct |
2 |
Correct |
1 ms |
4444 KB |
Output is correct |
3 |
Incorrect |
1 ms |
4444 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
30 ms |
15056 KB |
Output is correct |
2 |
Correct |
36 ms |
14956 KB |
Output is correct |
3 |
Correct |
21 ms |
11736 KB |
Output is correct |
4 |
Correct |
5 ms |
6236 KB |
Output is correct |
5 |
Correct |
4 ms |
5464 KB |
Output is correct |
6 |
Correct |
8 ms |
7004 KB |
Output is correct |
7 |
Incorrect |
1 ms |
4700 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |