#include<bits/stdc++.h>
using namespace std;
//#define LOCAL
#ifndef LOCAL
#include "swap.h"
#endif // LOCAL
const int N = 1e5 + 10;
int n, m;
vector<pair<int, int>> ad[N];
vector<int> co;
bool sub1 = 1, sub2 = 1;
int maxw = 0;
int par[N];
void init(int N, int M, vector<int> U, vector<int> V, vector<int> W) {
n = N; m = M;
for (int i = 0; i < m; ++i) {
int u = U[i], v = V[i], w = W[i];
co.push_back(w);
ad[u].push_back({v, w});
ad[v].push_back({u, w});
maxw = max(maxw, w);
if (ad[u].size() > 2 || ad[v].size() > 2) sub1 = 0;
if (u) {
sub2 = 0;
par[v] = w;
}
}
if (m != n - 1) sub2 = 0;
if (sub2) {
using pii = pair<int, int>;
sort(ad[0].begin(), ad[0].end(), [&] (pii a, pii b) {
return make_pair(a.second, a.first) < make_pair(b.second, b.first);
});
}
sort(co.begin(), co.end());
co.resize(unique(co.begin(), co.end()) - co.begin());
}
bool dd[N];
int special;
bool dfs1(int u, int pre, int cw) {
if (u == special) return 1;
bool ret = 0;
for (auto duck : ad[u]) {
int v, w; tie(v, w) = duck;
if (v == pre || w > cw) continue;
if (!dd[v]) {
dd[v] = 1;
ret |= dfs1(v, u, cw);
}
}
return ret;
}
bool dfs2(int u, int pre, int cw) {
int cnt = pre != -1;
bool ret = 0;
map<int, int> d;
for (auto duck : ad[u]) {
int v, w; tie(v, w) = duck;
if (v == pre || w > cw) continue;
if (dd[v] && !d[v]) return 1;
if (!dd[v]) {
dd[v] = d[v] = 1;
cnt += 1;
ret |= dfs2(v, u, cw);
}
}
return (cnt > 2 || ret);
}
int getMinimumFuelCapacity(int X, int Y) {
if (sub1 && m == n - 1) return -1;
if (sub1 && m == n) return maxw;
if (sub2) {
int it = 0;
int answer = max(par[X], par[Y]);
while (ad[0][it].first == X || ad[0][it].first == Y) it += 1;
answer = max(answer, ad[0][it].second);
if (!X || !Y) {
it += 1;
while (ad[0][it].first == X || ad[0][it].first == Y) it += 1;
answer = max(answer, ad[0][it].second);
}
return answer;
}
int l = 0, r = co.size() - 1;
int answer = -1;
special = Y;
while (l <= r) {
int mid = l + r >> 1;
bool ret = 1;
memset(dd, 0, sizeof dd); ret &= dfs1(X, -1, co[mid]);
memset(dd, 0, sizeof dd); ret &= dfs2(X, -1, co[mid]);
if (ret) r = mid - 1, answer = co[mid];
else l = mid + 1;
}
return answer;
}
#ifdef LOCAL
int32_t main() {
cin.tie(0)->sync_with_stdio(0);
if (fopen("duck.inp", "r")) {
freopen("duck.inp", "r", stdin);
freopen("duck.out", "w", stdout);
}
int n, m; cin >> n >> m;
vector<int> U(m), V(m), W(m);
for (int i = 0; i < m; ++i) cin >> U[i] >> V[i] >> W[i];
init(n, m, U, V, W);
int q; cin >> q;
for (int i = 1; i <= q; ++i) {
int X, Y; cin >> X >> Y;
cout << getMinimumFuelCapacity(X, Y) << '\n';
}
}
#endif // LOCAL
Compilation message
swap.cpp: In function 'bool dfs2(int, int, int)':
swap.cpp:74:20: warning: suggest parentheses around assignment used as truth value [-Wparentheses]
74 | dd[v] = d[v] = 1;
swap.cpp: In function 'int getMinimumFuelCapacity(int, int)':
swap.cpp:106:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
106 | int mid = l + r >> 1;
| ~~^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
2908 KB |
Output is correct |
2 |
Correct |
1 ms |
2908 KB |
Output is correct |
3 |
Correct |
1 ms |
2908 KB |
Output is correct |
4 |
Correct |
1 ms |
2908 KB |
Output is correct |
5 |
Correct |
1 ms |
3164 KB |
Output is correct |
6 |
Correct |
2 ms |
3164 KB |
Output is correct |
7 |
Correct |
2 ms |
3164 KB |
Output is correct |
8 |
Correct |
1 ms |
3120 KB |
Output is correct |
9 |
Correct |
30 ms |
8896 KB |
Output is correct |
10 |
Correct |
38 ms |
10092 KB |
Output is correct |
11 |
Correct |
36 ms |
9936 KB |
Output is correct |
12 |
Correct |
38 ms |
10184 KB |
Output is correct |
13 |
Correct |
41 ms |
10292 KB |
Output is correct |
14 |
Correct |
33 ms |
9136 KB |
Output is correct |
15 |
Correct |
77 ms |
12768 KB |
Output is correct |
16 |
Correct |
76 ms |
12664 KB |
Output is correct |
17 |
Correct |
81 ms |
13360 KB |
Output is correct |
18 |
Correct |
88 ms |
13544 KB |
Output is correct |
19 |
Correct |
41 ms |
8756 KB |
Output is correct |
20 |
Correct |
87 ms |
14080 KB |
Output is correct |
21 |
Correct |
81 ms |
14132 KB |
Output is correct |
22 |
Correct |
92 ms |
14900 KB |
Output is correct |
23 |
Correct |
83 ms |
14768 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
2908 KB |
Output is correct |
2 |
Correct |
1 ms |
2908 KB |
Output is correct |
3 |
Incorrect |
82 ms |
13280 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
2908 KB |
Output is correct |
2 |
Correct |
1 ms |
2908 KB |
Output is correct |
3 |
Correct |
1 ms |
2908 KB |
Output is correct |
4 |
Correct |
1 ms |
2908 KB |
Output is correct |
5 |
Correct |
1 ms |
3164 KB |
Output is correct |
6 |
Correct |
2 ms |
3164 KB |
Output is correct |
7 |
Correct |
2 ms |
3164 KB |
Output is correct |
8 |
Correct |
1 ms |
3120 KB |
Output is correct |
9 |
Correct |
1 ms |
3164 KB |
Output is correct |
10 |
Correct |
2 ms |
3164 KB |
Output is correct |
11 |
Correct |
2 ms |
3160 KB |
Output is correct |
12 |
Correct |
2 ms |
3164 KB |
Output is correct |
13 |
Correct |
2 ms |
3164 KB |
Output is correct |
14 |
Correct |
3 ms |
3160 KB |
Output is correct |
15 |
Correct |
2 ms |
3164 KB |
Output is correct |
16 |
Correct |
2 ms |
3164 KB |
Output is correct |
17 |
Correct |
3 ms |
3420 KB |
Output is correct |
18 |
Correct |
3 ms |
3164 KB |
Output is correct |
19 |
Correct |
3 ms |
3332 KB |
Output is correct |
20 |
Correct |
2 ms |
3376 KB |
Output is correct |
21 |
Correct |
3 ms |
3164 KB |
Output is correct |
22 |
Correct |
3 ms |
3120 KB |
Output is correct |
23 |
Correct |
3 ms |
3164 KB |
Output is correct |
24 |
Correct |
4 ms |
3420 KB |
Output is correct |
25 |
Correct |
5 ms |
3164 KB |
Output is correct |
26 |
Correct |
3 ms |
3420 KB |
Output is correct |
27 |
Correct |
3 ms |
3164 KB |
Output is correct |
28 |
Correct |
4 ms |
3420 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
3164 KB |
Output is correct |
2 |
Correct |
1 ms |
2908 KB |
Output is correct |
3 |
Correct |
1 ms |
2908 KB |
Output is correct |
4 |
Correct |
1 ms |
2908 KB |
Output is correct |
5 |
Correct |
1 ms |
2908 KB |
Output is correct |
6 |
Correct |
1 ms |
3164 KB |
Output is correct |
7 |
Correct |
2 ms |
3164 KB |
Output is correct |
8 |
Correct |
2 ms |
3164 KB |
Output is correct |
9 |
Correct |
1 ms |
3120 KB |
Output is correct |
10 |
Correct |
30 ms |
8896 KB |
Output is correct |
11 |
Correct |
38 ms |
10092 KB |
Output is correct |
12 |
Correct |
36 ms |
9936 KB |
Output is correct |
13 |
Correct |
38 ms |
10184 KB |
Output is correct |
14 |
Correct |
41 ms |
10292 KB |
Output is correct |
15 |
Correct |
2 ms |
3164 KB |
Output is correct |
16 |
Correct |
2 ms |
3160 KB |
Output is correct |
17 |
Correct |
2 ms |
3164 KB |
Output is correct |
18 |
Correct |
2 ms |
3164 KB |
Output is correct |
19 |
Correct |
3 ms |
3160 KB |
Output is correct |
20 |
Correct |
2 ms |
3164 KB |
Output is correct |
21 |
Correct |
2 ms |
3164 KB |
Output is correct |
22 |
Correct |
3 ms |
3420 KB |
Output is correct |
23 |
Correct |
3 ms |
3164 KB |
Output is correct |
24 |
Correct |
3 ms |
3332 KB |
Output is correct |
25 |
Correct |
2 ms |
3376 KB |
Output is correct |
26 |
Correct |
3 ms |
3164 KB |
Output is correct |
27 |
Correct |
3 ms |
3120 KB |
Output is correct |
28 |
Correct |
3 ms |
3164 KB |
Output is correct |
29 |
Correct |
4 ms |
3420 KB |
Output is correct |
30 |
Correct |
5 ms |
3164 KB |
Output is correct |
31 |
Correct |
3 ms |
3420 KB |
Output is correct |
32 |
Correct |
3 ms |
3164 KB |
Output is correct |
33 |
Correct |
4 ms |
3420 KB |
Output is correct |
34 |
Correct |
11 ms |
4444 KB |
Output is correct |
35 |
Correct |
194 ms |
25456 KB |
Output is correct |
36 |
Correct |
152 ms |
18380 KB |
Output is correct |
37 |
Correct |
147 ms |
12356 KB |
Output is correct |
38 |
Correct |
266 ms |
10556 KB |
Output is correct |
39 |
Correct |
174 ms |
9840 KB |
Output is correct |
40 |
Correct |
272 ms |
9676 KB |
Output is correct |
41 |
Correct |
201 ms |
20288 KB |
Output is correct |
42 |
Correct |
225 ms |
22836 KB |
Output is correct |
43 |
Correct |
555 ms |
23440 KB |
Output is correct |
44 |
Correct |
288 ms |
10696 KB |
Output is correct |
45 |
Correct |
695 ms |
18232 KB |
Output is correct |
46 |
Correct |
310 ms |
20820 KB |
Output is correct |
47 |
Correct |
321 ms |
10288 KB |
Output is correct |
48 |
Correct |
409 ms |
10940 KB |
Output is correct |
49 |
Correct |
92 ms |
13860 KB |
Output is correct |
50 |
Correct |
124 ms |
11208 KB |
Output is correct |
51 |
Correct |
576 ms |
16036 KB |
Output is correct |
52 |
Correct |
1118 ms |
23552 KB |
Output is correct |
53 |
Correct |
311 ms |
15600 KB |
Output is correct |
54 |
Correct |
1673 ms |
30144 KB |
Output is correct |
55 |
Correct |
423 ms |
18316 KB |
Output is correct |
56 |
Correct |
1195 ms |
20936 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
2908 KB |
Output is correct |
2 |
Correct |
1 ms |
2908 KB |
Output is correct |
3 |
Correct |
1 ms |
2908 KB |
Output is correct |
4 |
Correct |
1 ms |
2908 KB |
Output is correct |
5 |
Correct |
1 ms |
3164 KB |
Output is correct |
6 |
Correct |
2 ms |
3164 KB |
Output is correct |
7 |
Correct |
2 ms |
3164 KB |
Output is correct |
8 |
Correct |
1 ms |
3120 KB |
Output is correct |
9 |
Correct |
30 ms |
8896 KB |
Output is correct |
10 |
Correct |
38 ms |
10092 KB |
Output is correct |
11 |
Correct |
36 ms |
9936 KB |
Output is correct |
12 |
Correct |
38 ms |
10184 KB |
Output is correct |
13 |
Correct |
41 ms |
10292 KB |
Output is correct |
14 |
Correct |
33 ms |
9136 KB |
Output is correct |
15 |
Correct |
77 ms |
12768 KB |
Output is correct |
16 |
Correct |
76 ms |
12664 KB |
Output is correct |
17 |
Correct |
81 ms |
13360 KB |
Output is correct |
18 |
Correct |
88 ms |
13544 KB |
Output is correct |
19 |
Incorrect |
82 ms |
13280 KB |
Output isn't correct |
20 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
3164 KB |
Output is correct |
2 |
Correct |
1 ms |
2908 KB |
Output is correct |
3 |
Correct |
1 ms |
2908 KB |
Output is correct |
4 |
Correct |
1 ms |
2908 KB |
Output is correct |
5 |
Correct |
1 ms |
2908 KB |
Output is correct |
6 |
Correct |
1 ms |
3164 KB |
Output is correct |
7 |
Correct |
2 ms |
3164 KB |
Output is correct |
8 |
Correct |
2 ms |
3164 KB |
Output is correct |
9 |
Correct |
1 ms |
3120 KB |
Output is correct |
10 |
Correct |
30 ms |
8896 KB |
Output is correct |
11 |
Correct |
38 ms |
10092 KB |
Output is correct |
12 |
Correct |
36 ms |
9936 KB |
Output is correct |
13 |
Correct |
38 ms |
10184 KB |
Output is correct |
14 |
Correct |
41 ms |
10292 KB |
Output is correct |
15 |
Correct |
33 ms |
9136 KB |
Output is correct |
16 |
Correct |
77 ms |
12768 KB |
Output is correct |
17 |
Correct |
76 ms |
12664 KB |
Output is correct |
18 |
Correct |
81 ms |
13360 KB |
Output is correct |
19 |
Correct |
88 ms |
13544 KB |
Output is correct |
20 |
Incorrect |
82 ms |
13280 KB |
Output isn't correct |
21 |
Halted |
0 ms |
0 KB |
- |