#include <bits/stdc++.h>
using namespace std;
#include "dreaming.h"
#define sp " "
#define endl "\n"
#define pb push_back
#define pii pair<int, int>
#define st first
#define nd second
#define fileio() freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout)
#define fastio() cin.tie(0), ios_base::sync_with_stdio(0)
#define mid (l + r) / 2
#define LL node * 2
#define RR node * 2 + 1
#define ll long long
#define MAXN 300005
const int modulo = 1e9 + 7;
const ll INF = 2e18 + 7;
int maks[MAXN], sub_maks[MAXN], vis[MAXN];
vector<pii> adj[MAXN];
int dfs2(int node, int root){
int ans = 0;
for (auto i : adj[node]){
if (i.st == root) continue;
ans = max(ans, dfs2(i.st, node) + i.nd);
}
return sub_maks[node] = ans;
}
int dfs(int node, int root, int m){
vis[node] = 1;
maks[node] = max(m, sub_maks[node]);
int ans = maks[node];
vector<array<int, 3>> v;
for (auto i : adj[node]){
if (i.st == root) continue;
v.pb({sub_maks[i.st] + i.nd, i.st, i.nd});
}
sort(v.begin(), v.end());
for (int i = 0; i < v.size(); i++){
int to = v[i][1], w = v[i][2];
int add = m;
if (i + 1 < v.size()) add = max(add, v.back()[0]);
else add = max(add, v[v.size() - 2][0]);
ans = min(ans, dfs(to, node, add + w));
}
return ans;
}
int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
for (int i = 0; i < M; i++){
A[i]++, B[i]++;
adj[A[i]].pb({B[i], T[i]});
adj[B[i]].pb({A[i], T[i]});
}
priority_queue<int> q;
for (int i = 1; i <= N; i++){
if (vis[i]) continue;
int tmp = dfs2(i, 0);
int to = dfs(i, 0, 0);
q.push(to);
}
int ans = 0;
while(!q.empty()){
int a = q.top();
q.pop();
if (q.empty()) break;
int b = q.top();
q.pop();
int neww = max(a, b + L);
ans = max(ans, a + b + L);
q.push(neww);
}
return ans;
}
Compilation message
dreaming.cpp: In function 'int dfs(int, int, int)':
dreaming.cpp:48:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::array<int, 3> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
48 | for (int i = 0; i < v.size(); i++){
| ~~^~~~~~~~~~
dreaming.cpp:51:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::array<int, 3> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
51 | if (i + 1 < v.size()) add = max(add, v.back()[0]);
| ~~~~~~^~~~~~~~~~
dreaming.cpp: In function 'int travelTime(int, int, int, int*, int*, int*)':
dreaming.cpp:69:10: warning: unused variable 'tmp' [-Wunused-variable]
69 | int tmp = dfs2(i, 0);
| ^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
34 ms |
23132 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
10584 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
34 ms |
23132 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
18 ms |
13912 KB |
Output is correct |
2 |
Correct |
21 ms |
13916 KB |
Output is correct |
3 |
Correct |
21 ms |
13916 KB |
Output is correct |
4 |
Correct |
18 ms |
13912 KB |
Output is correct |
5 |
Correct |
20 ms |
13912 KB |
Output is correct |
6 |
Correct |
19 ms |
14208 KB |
Output is correct |
7 |
Correct |
21 ms |
14076 KB |
Output is correct |
8 |
Correct |
19 ms |
13896 KB |
Output is correct |
9 |
Correct |
19 ms |
13784 KB |
Output is correct |
10 |
Correct |
21 ms |
14040 KB |
Output is correct |
11 |
Correct |
2 ms |
10588 KB |
Output is correct |
12 |
Correct |
8 ms |
11736 KB |
Output is correct |
13 |
Correct |
8 ms |
11736 KB |
Output is correct |
14 |
Correct |
8 ms |
11736 KB |
Output is correct |
15 |
Correct |
9 ms |
11736 KB |
Output is correct |
16 |
Correct |
8 ms |
11736 KB |
Output is correct |
17 |
Correct |
8 ms |
11480 KB |
Output is correct |
18 |
Correct |
12 ms |
11736 KB |
Output is correct |
19 |
Correct |
8 ms |
11736 KB |
Output is correct |
20 |
Correct |
1 ms |
10588 KB |
Output is correct |
21 |
Correct |
1 ms |
10588 KB |
Output is correct |
22 |
Correct |
1 ms |
10588 KB |
Output is correct |
23 |
Correct |
8 ms |
11736 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
10584 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
34 ms |
23132 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |